diff --git a/tests/unit/test_oceanbase_peewee.py b/test/unit_test/utils/test_oceanbase_peewee.py similarity index 98% rename from tests/unit/test_oceanbase_peewee.py rename to test/unit_test/utils/test_oceanbase_peewee.py index 7ff0ef2d2..6ddee9850 100644 --- a/tests/unit/test_oceanbase_peewee.py +++ b/test/unit_test/utils/test_oceanbase_peewee.py @@ -71,7 +71,7 @@ class TestOceanBaseDatabase: def test_database_lock_enum_values(self): """Test DatabaseLock enum has all expected values.""" expected = {'MYSQL', 'OCEANBASE', 'POSTGRES'} - actual = {e.name for e in DatabaseLock} + actual = set(DatabaseLock.__members__.keys()) assert expected.issubset(actual), f"Missing: {expected - actual}"