[cli] Add ability to delete remote object

2019-12-27 22:53:49 -05:00
parent bfe870be65
commit f78ebc22ad
4 changed files with 43 additions and 10 deletions
--- a/tests/test_hf_api.py
+++ b/tests/test_hf_api.py
@@ -60,6 +60,11 @@ class HfApiEndpointsTest(HfApiCommonTest):
        """
        cls._token = cls._api.login(username=USER, password=PASS)

+    @classmethod
+    def tearDownClass(cls):
+        for FILE_KEY, FILE_PATH in FILES:
+            cls._api.delete_obj(token=cls._token, filename=FILE_KEY)
+
    def test_whoami(self):
        user = self._api.whoami(token=self._token)
        self.assertEqual(user, USER)