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/")