update mode:cloud-only (#15)

SwanHubX · Jul 17, 2024 · 8540d8f · 8540d8f
1 parent 11a6867
commit 8540d8f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -9,7 +9,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "swankit"
-version = "0.1.1b1"
+version = "0.1.1b2"
 dynamic = ["readme"]
 description = "Base toolkit for SwanLab"
 license = "Apache-2.0"

diff --git a/swankit/env.py b/swankit/env.py
@@ -21,7 +21,7 @@ class SwanLabMode(Enum):
 
     DISABLED = "disabled"
     CLOUD = "cloud"
-    # CLOUD_ONLY = "cloud-only"
+    CLOUD_ONLY = "cloud-only"
     LOCAL = "local"
 
     @classmethod

diff --git a/test/unit/test_env.py b/test/unit/test_env.py
@@ -23,6 +23,15 @@ def test_create_time():
     assert d.tzinfo == datetime.timezone.utc
 
 
+def test_list_mode():
+    ms = E.SwanLabMode.list()
+    assert len(ms) == 4
+    assert "disabled" in ms
+    assert "cloud" in ms
+    assert "cloud-only" in ms
+    assert "local" in ms
+
+
 class TestGetFolder:
     """
     测试获取全局保存文件夹