def testCreateModelInfo(self): did_raise_value_error = False try: retrain.create_model_info('no_such_model_name') except ValueError: did_raise_value_error = True self.assertTrue(did_raise_value_error) model_info = retrain.create_model_info('inception_v3') self.assertIsNotNone(model_info) self.assertEqual(299, model_info['input_width'])
def testCreateModelInfoQuantized(self): # Test for mobilenet_quantized model_info = retrain.create_model_info('mobilenet_1.0_224') self.assertIsNotNone(model_info) self.assertEqual(224, model_info['input_width'])