Fix test

rapidsai · Nov 12, 2024 · 9e3ef17 · 9e3ef17
1 parent 2a96d74
commit 9e3ef17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/kvikio/tests/test_defaults.py b/python/kvikio/tests/test_defaults.py
@@ -16,9 +16,9 @@ def test_compat_mode():
 
     before = kvikio.defaults.compat_mode()
     with kvikio.defaults.set_compat_mode(kvikio.CompatMode.ON):
-        assert kvikio.defaults.compat_mode()
+        assert kvikio.defaults.compat_mode() == kvikio.CompatMode.ON
         kvikio.defaults.compat_mode_reset(kvikio.CompatMode.OFF)
-        assert not kvikio.defaults.compat_mode()
+        assert kvikio.defaults.compat_mode() == kvikio.CompatMode.OFF
     assert before == kvikio.defaults.compat_mode()