Update BMS tests to assert correct model was found

[TheLemonPig]

Currently BMS tests just confirm that the model runs. You can run tests on verbose to manually check to see if they match if you want. In order to add these asserts and have the tests reliably pass when they should will need to:

• increase the number of epochs that BMS trains for in the tests (this will probably increase how long the tests take by a few minutes)
• set a seed, because currently BMS does not seem to give the same output (although I thought this was the case in the past) - I guess the seed was accidentally removed at some point