update

pyg-team · Sep 2, 2022 · 9d1c9ea · 9d1c9ea
1 parent 8142cc9
commit 9d1c9ea
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/test_experimental.py b/test/test_experimental.py
@@ -14,12 +14,15 @@ def test_experimental_mode(options):
         assert is_experimental_mode_enabled(options) is True
     assert is_experimental_mode_enabled(options) is False
 
-    assert is_experimental_mode_enabled(options) is False
     with set_experimental_mode(True, options):
         assert is_experimental_mode_enabled(options) is True
     assert is_experimental_mode_enabled(options) is False
 
-    assert is_experimental_mode_enabled(options) is False
     with set_experimental_mode(False, options):
         assert is_experimental_mode_enabled(options) is False
     assert is_experimental_mode_enabled(options) is False
+
+    set_experimental_mode(True, options)
+    assert is_experimental_mode_enabled(options) is True
+    set_experimental_mode(False, options)
+    assert is_experimental_mode_enabled(options) is False