Write tests to check the model predictions on simple datasets
This ensures that the training/prediction callbacks of the surrogates are working fine.
Related: #11
This ensures that the training/prediction callbacks of the surrogates are working fine.
Related: #11