Class TestKaggleModuleResolver

tests/test_kaggle_module_resolver.py:81–117

            self.wfile.write(bytes(f"Unhandled path: {self.path}", "utf-8"))

class TestKaggleModuleResolver(unittest.TestCase):
    def test_kaggle_resolver_long_url_succeeds(self):
        model_url = "https://kaggle.com/models/foo/foomodule/frameworks/TensorFlow2/variations/barvar/versions/2"
        with create_test_server(KaggleJwtHandler) as addr:
            test_inputs = tf.ones([1,4])
            layer = hub.KerasLayer(model_url)
            self.assertEqual([1, 1], layer(test_inputs).shape)
        # Delete the files that were created in KaggleJwtHandler's do_POST method
        os.unlink(os.path.join(MOUNT_PATH, "foomodule/tensorflow2/barvar/2"))
        os.rmdir(os.path.dirname(os.path.join(MOUNT_PATH, "foomodule/tensorflow2/barvar/2")))

    def test_kaggle_resolver_short_url_succeeds(self):
        model_url = "https://kaggle.com/models/foo/foomodule/TensorFlow2/barvar/2"
        with create_test_server(KaggleJwtHandler) as addr:
            test_inputs = tf.ones([1,4])
            layer = hub.KerasLayer(model_url)
            self.assertEqual([1, 1], layer(test_inputs).shape)
        # Delete the files that were created in KaggleJwtHandler's do_POST method
        os.unlink(os.path.join(MOUNT_PATH, "foomodule/tensorflow2/barvar/2"))
        os.rmdir(os.path.dirname(os.path.join(MOUNT_PATH, "foomodule/tensorflow2/barvar/2")))

    def test_kaggle_resolver_not_attached_throws(self):
        with create_test_server(KaggleJwtHandler) as addr:
            with self.assertRaises(BackendError):
                hub.KerasLayer("https://kaggle.com/models/foo/unknown/frameworks/TensorFlow2/variations/barvar/versions/2")

    def test_http_resolver_succeeds(self):
        with create_test_server(HubHTTPHandler) as addr:
            test_inputs = tf.ones([1,4])
            layer = hub.KerasLayer(f'{addr}/model.tar.gz')
            self.assertEqual([1, 1], layer(test_inputs).shape)

    def test_local_path_resolver_succeeds(self):
        test_inputs = tf.ones([1,4])
        layer = hub.KerasLayer('/input/tests/data/saved_model')
        
        self.assertEqual([1, 1], layer(test_inputs).shape)