add tests

README.md

@@ -8,6 +8,12 @@ app_file: app.py
 pinned: false
 ---
 
+# Tests
+```
+pytest -v --cov=. --cov-report term-missing tests.py
+```
+
+
 # Configuration
 
 `title`: _string_  

tests.py

@@ -0,0 +1,41 @@
+import pytest
+from app import SpaceBuilder
+
+
+class TestUnit:
+    class TestSpaceBuilder:
+        def test_split_space_names(self):
+            input = (
+                f"nielsr/LayoutLMv2-FUNSD"
+                f"\nvalhalla/glide-text2im"
+                f"\n"
+                f"\n    "
+                f"\n   "
+                f"\nvalhalla/glide-text2im"
+                f"\n  "
+                f"\nvalhalla/glide-text2im"
+                f"\n"
+            )
+            expected_output = (
+                ["spaces/nielsr/LayoutLMv2-FUNSD", "spaces/valhalla/glide-text2im", "spaces/valhalla/glide-text2im", "spaces/valhalla/glide-text2im"]
+            )
+            assert expected_output == SpaceBuilder.split_space_names(input)
+
+        def test_check_space_name_availability(self):
+            random_token = "123123"
+            random_space_name = "123123"
+            assert SpaceBuilder.check_space_name_availability(random_token, random_space_name) is False
+
+        def test_load_and_check_spaces_1(self):
+            input = (
+                f"nielsr/LayoutLMv2-FUNSD"
+                f"\nvalhalla/glide-text2im"
+            )
+            assert SpaceBuilder.load_and_check_spaces(names=input) is False
+
+        def test_load_and_check_spaces_2(self):
+            input = (
+                f"valhalla/glide-text2im"
+                f"\nvalhalla/glide-text2im"
+            )
+            assert SpaceBuilder.load_and_check_spaces(names=input) is True