Add tests for prediction code

Closes #126 (closed)

changed milestone to %DAN-P4: Improve data loading and preprocessing

added P1 label

requested review from @starride

assigned to @melodie.boillet

Looks great!

approved this merge request

merged