SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[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
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__" PASS = "__DUMMY_TRANSFORMERS_PASS__"
FILES = [ 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"), 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"), 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): def test_presign_invalid_org(self):
with self.assertRaises(HTTPError): 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): 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) self.assertIsInstance(urls, PresignedUrl)
def test_presign(self): def test_presign(self):