[model upload] Support for organizations

This commit is contained in:
Julien Chaumond committed 2020-03-09 17:33:57 -04:00
1 parent 525b6b1c54
commit cbf8f5d32b
3 files changed
+50 -18

No files matched your search

+10 -1
View File
@@ -67,8 +67,17 @@ class HfApiEndpointsTest(HfApiCommonTest):
cls._api.delete_obj(token=cls._token, filename=FILE_KEY)
def test_whoami(self):
user = self._api.whoami(token=self._token)
user, orgs = self._api.whoami(token=self._token)
self.assertEqual(user, USER)
self.assertIsInstance(orgs, list)
def test_presign_invalid_org(self):
with self.assertRaises(HTTPError):
_ = self._api.presign(token=self._token, filename="fake_org.txt", organization="fake")
def test_presign_valid_org(self):
urls = self._api.presign(token=self._token, filename="valid_org.txt", organization="valid_org")
self.assertIsInstance(urls, PresignedUrl)
def test_presign(self):
for FILE_KEY, FILE_PATH in FILES: