feat: support for samples (input/output) routes. (#7)

Author: raul-casallas

modzy/models.py

@@ -107,12 +107,44 @@ def get_version(self, model, version):
             ApiError: A subclass of ApiError will be raised if the API returns an error status,
                 or the client is unable to connect.
         """
-        self.logger.debug("getting versopm model %s version %s", model, version)
+        self.logger.debug("getting version model %s version %s", model, version)
         modelId = Model._coerce_identifier(model)
         versionId = ModelVersion._coerce_identifier(version)
         json_obj = self._api_client.http.get('{}/{}/versions/{}'.format(self._base_route, modelId, versionId))
         return ModelVersion(json_obj, self._api_client)
 
+    def get_version_input_sample(self, model, version):
+        """Gets the input sample associated with the model and version provided.
+
+        Returns:
+            String: A json string with the input sample
+
+        Raises:
+            ApiError: A subclass of ApiError will be raised if the API returns an error status,
+                or the client is unable to connect.
+        """
+        self.logger.debug("getting input sample: model %s version %s", model, version)
+        modelId = Model._coerce_identifier(model)
+        versionId = ModelVersion._coerce_identifier(version)
+        json_obj = self._api_client.http.get('{}/{}/versions/{}/sample-input'.format(self._base_route, modelId, versionId))
+        return json_obj
+
+    def get_version_output_sample(self, model, version):
+        """Gets the output sample associated with the model and version provided.
+
+        Returns:
+            String: A json string with the output sample
+
+        Raises:
+            ApiError: A subclass of ApiError will be raised if the API returns an error status,
+                or the client is unable to connect.
+        """
+        self.logger.debug("getting output sample: model %s version %s", model, version)
+        modelId = Model._coerce_identifier(model)
+        versionId = ModelVersion._coerce_identifier(version)
+        json_obj = self._api_client.http.get('{}/{}/versions/{}/sample-output'.format(self._base_route, modelId, versionId))
+        return json_obj
+
     def get_all(self):
         """Gets a list of all `Model` instances.
 

tests/test_models.py

@@ -102,3 +102,21 @@ def test_get_model_versions(client, logger):
         logger.debug("version: %s", version)
         assert version.version
     assert len(version)  # just going to assume there should be some versions
+
+
+def test_get_model_version(client, logger):
+    version = client.models.get_version(MODEL_ID, '0.0.27')
+    logger.debug("version: %s", version)
+    assert version.version
+
+
+def test_get_model_version_input_sample(client, logger):
+    input_sample = client.models.get_version_input_sample(MODEL_ID, '0.0.27')
+    logger.debug("version: %s", input_sample)
+    assert input_sample
+
+
+def test_get_model_version_output_sample(client, logger):
+    output_sample = client.models.get_version_output_sample(MODEL_ID, '0.0.27')
+    logger.debug("version: %s", output_sample)
+    assert output_sample