def testValidateInputPathForKerasSavedModel(self): self.assertNotEqual(True, wizard.validate_input_path( self._tmp_dir, 'keras_saved_model')) self._create_keras_saved_model() save_dir = os.path.join(self._tmp_dir, SAVED_MODEL_DIR) self.assertEqual(True, wizard.validate_input_path( save_dir, 'keras_saved_model'))
def testValidateInputPathForKerasModel(self): self.assertNotEqual(True, wizard.validate_input_path(self._tmp_dir, 'keras')) self._create_hd5_file() save_dir = os.path.join(self._tmp_dir, HD5_FILE_NAME) self.assertEqual(True, wizard.validate_input_path( save_dir, 'keras'))
def testValidateInputPathForTFHub(self): self.assertNotEqual( True, wizard.validate_input_path(self._tmp_dir, 'tf_hub')) self.assertEqual( True, wizard.validate_input_path("https://tfhub.dev/mobilenet", 'tf_hub'))
def testValidateInputPathForLayersModel(self): self.assertNotEqual(True, wizard.validate_input_path(self._tmp_dir, 'keras')) self._create_layers_model() save_dir = os.path.join(self._tmp_dir) self.assertEqual(True, wizard.validate_input_path( save_dir, 'tfjs_layers_model')) save_dir = os.path.join(self._tmp_dir, 'model.json') self.assertEqual(True, wizard.validate_input_path( save_dir, 'tfjs_layers_model'))