Merge pull request #427 from watson-developer-cloud/codegen/visual-recognition

feat(core_ml): Add get_core_ml_model in VR

Author: germanattanasio

.gitignore

@@ -64,3 +64,4 @@ test/__init__.py
 *~
 
 .sfdx/tools/apex.db
+.pytest_cache/

test/integration/test_visual_recognition.py

@@ -19,14 +19,15 @@ def setUp(self):
             'X-Watson-Test':
             '1'
         })
+        self.classifier_id = 'CarsvsTrucks_931077904'
 
     def test_classify(self):
         car_path = join(dirname(__file__), '../../resources/cars.zip')
         with open(car_path, 'rb') as images_file:
             parameters = json.dumps({
                 'threshold':
                 0.1,
-                'classifier_ids': ['CarsvsTrucks_931077904', 'default']
+                'classifier_ids': [self.classifier_id, 'default']
             })
             car_results = self.visual_recognition.classify(
                 images_file=images_file, parameters=parameters)
@@ -46,7 +47,8 @@ def test_custom_classifier(self):
             classifier = self.visual_recognition.create_classifier(
                 'Cars vs Trucks',
                 cars_positive_examples=cars,
-                negative_examples=trucks)
+                negative_examples=trucks,
+                )
 
         assert classifier is not None
 
@@ -58,3 +60,7 @@ def test_custom_classifier(self):
         assert classifiers is not None
 
         output = self.visual_recognition.delete_classifier(classifier_id)
+
+    def test_core_ml_model(self):
+        core_ml_model = self.visual_recognition.get_core_ml_model(self.classifier_id)
+        assert core_ml_model.ok