def testSaveModelOK(self):
dann = Dry()
model = dann.train_model(bval1s)
dann.save_model(model, os.path.join(testingdir, "testmodel"))
self.assertTrue(os.path.isfile(os.path.join(testingdir, "testmodel")))
def testLoadModelOK(self):
dann = Dry()
model = dann.train_model(bval1s)
dann.save_model(model, os.path.join(testingdir, "testmodel"))
model2 = dann.load_model(os.path.join(testingdir, "testmodel"))
self.assertIsInstance(model2, Sequential)
def testSaveModelNoModel(self):
dann = Dry()
with self.assertRaises(ValueError):
dann.save_model(None, "dry/data/")