Add Unit testing

[ibrahimsharaf]

Add unit tests to trigger the model training and inference then make sure the performance doesn't worsen below a certain threshold (80%?)

Depends on #14