pytest for writing test functions

[emilymuller1991]

Create tests for the train/ repository functions. Create new files for function. Add in test_images/ Use mock to test components within code.