Add unit test

Author: lucyhe

tests/test_kaggle_module_resolver.py

@@ -79,12 +79,18 @@ def do_POST(self):
             self.wfile.write(bytes(f"Unhandled path: {self.path}", "utf-8"))
 
 class TestKaggleModuleResolver(unittest.TestCase):
-    def test_kaggle_resolver_succeeds(self):        
+    def test_kaggle_resolver_long_url_succeeds(self):        
         with create_test_server(KaggleJwtHandler) as addr:
             test_inputs = tf.ones([1,4])
             layer = hub.KerasLayer("https://kaggle.com/models/foo/foomodule/frameworks/TensorFlow2/variations/barvar/versions/2")
             self.assertEqual([1, 1], layer(test_inputs).shape)
 
+    def test_kaggle_resolver_short_url_succeeds(self):        
+        with create_test_server(KaggleJwtHandler) as addr:
+            test_inputs = tf.ones([1,4])
+            layer = hub.KerasLayer("https://kaggle.com/models/foo/foomodule/TensorFlow2/barvar/2")
+            self.assertEqual([1, 1], layer(test_inputs).shape)
+
     def test_kaggle_resolver_not_attached_throws(self):
         with create_test_server(KaggleJwtHandler) as addr:
             with self.assertRaises(BackendError):