FlashCode-Lab commited on
Commit
ca7b166
·
verified ·
1 Parent(s): e56f770

Create tests/test_deploy_to_hf_space.py

Browse files
Files changed (1) hide show
  1. tests/test_deploy_to_hf_space.py +31 -0
tests/test_deploy_to_hf_space.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import unittest
2
+
3
+ from scripts.deploy_to_hf_space import build_create_repo_payload, resolve_publish_files, resolve_token
4
+
5
+
6
+ class DeployScriptTests(unittest.TestCase):
7
+ def test_build_create_repo_payload(self):
8
+ payload = build_create_repo_payload(
9
+ space_name="demo-space",
10
+ organization="demo-org",
11
+ private=True,
12
+ sdk="gradio",
13
+ )
14
+ self.assertEqual(payload["type"], "space")
15
+ self.assertEqual(payload["name"], "demo-space")
16
+ self.assertEqual(payload["organization"], "demo-org")
17
+ self.assertTrue(payload["private"])
18
+ self.assertEqual(payload["sdk"], "gradio")
19
+
20
+ def test_resolve_publish_files_deduplicate(self):
21
+ files = resolve_publish_files(["README.md", "assets/logo.png", "assets/logo.png"])
22
+ self.assertIn("index.html", files)
23
+ self.assertIn("assets/logo.png", files)
24
+ self.assertEqual(files.count("assets/logo.png"), 1)
25
+
26
+ def test_resolve_token_priority(self):
27
+ self.assertEqual(resolve_token("abc"), "abc")
28
+
29
+
30
+ if __name__ == "__main__":
31
+ unittest.main()