* added correctness tests for classification models
* refactored tests for extensibility & usability
* flake8 fixes