From 9e3ef1745a1c0b4a7e54ecb3a7e25c42feedcf42 Mon Sep 17 00:00:00 2001 From: Tianyu Liu Date: Tue, 12 Nov 2024 09:56:50 -0500 Subject: [PATCH] Fix test --- python/kvikio/tests/test_defaults.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/kvikio/tests/test_defaults.py b/python/kvikio/tests/test_defaults.py index 24cdb4c75c..bd448c2f84 100644 --- 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()