saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[ci] Align test_hf_api.py with API change

Browse source
This commit is contained in:
Julien Chaumond 2020-04-30 12:06:01 -04:00
parent 7f9193ef09
commit 452dd0e4d9
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/test_hf_api.py
View file

@ -28,11 +28,11 @@ USER = "__DUMMY_TRANSFORMERS_USER__"
PASS = "__DUMMY_TRANSFORMERS_PASS__"
FILES = [
(
"Test-{}.txt".format(int(time.time())),
"nested/Test-{}.txt".format(int(time.time())),
os.path.join(os.path.dirname(os.path.abspath(__file__)), "fixtures/input.txt"),
),
(
"yoyo {}.txt".format(int(time.time())), # space is intentional
"nested/yoyo {}.txt".format(int(time.time())), # space is intentional
os.path.join(os.path.dirname(os.path.abspath(__file__)), "fixtures/empty.txt"),
),
]
@ -73,10 +73,10 @@ class HfApiEndpointsTest(HfApiCommonTest):
def test_presign_invalid_org(self):
with self.assertRaises(HTTPError):
_ = self._api.presign(token=self._token, filename="fake_org.txt", organization="fake")
_ = self._api.presign(token=self._token, filename="nested/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")
urls = self._api.presign(token=self._token, filename="nested/valid_org.txt", organization="valid_org")
self.assertIsInstance(urls, PresignedUrl)
def test_presign(self):