Adding test for external uri

bmaguireibm · bmaguireibm · commit 35b8913f5456 · 2018-08-10T19:07:30.000+01:00
diff --git a/test/tests/commons/loadWeightMap.test.ts b/test/tests/commons/loadWeightMap.test.ts
@@ -59,6 +59,14 @@ describe('loadWeightMap', () => {
       expect(result.modelBaseUri).toEqual('/path/to/modelfiles')
     })
 
+    it('returns correct uris, given external path', () => {
+      const uri = 'https://example.com/path/to/modelfiles/';
+      const result = getModelUris(uri, FAKE_DEFAULT_MODEL_NAME)
+
+      expect(result.manifestUri).toEqual(uri + 'model-weights_manifest.json')
+      expect(result.modelBaseUri).toEqual(uri)
+    })
+
   })
 
 })
