write a unit test for the model initialization step

[alexatartaglini]

use pytest library. test should consist of generating a fake image, passing it through the model, and checking that the feature vector size is what we expect