Tests quick fix: tests/ops/models/test_connectionconfig.py (#5672)

ethyca · Jan 15, 2025 · b3cb8d0 · b3cb8d0
1 parent 72cd9b3
commit b3cb8d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/ops/models/test_connectionconfig.py b/tests/ops/models/test_connectionconfig.py
@@ -146,15 +146,15 @@ def test_connection_type_human_readable(self):
             connection.human_readable  # Makes sure all ConnectionTypes have been added to human_readable mapping
 
     def test_connection_type_human_readable_invalid(self):
-        with pytest.raises(NotImplementedError):
+        with pytest.raises(ValueError):
             ConnectionType("nonmapped_type").human_readable
 
     def test_connection_type_system_type(self):
         for connection in ConnectionType:
             connection.system_type  # Makes sure all ConnectionTypes have been added to system_type mapping
 
     def test_connection_type_system_type_invalid(self):
-        with pytest.raises(NotImplementedError):
+        with pytest.raises(ValueError):
             ConnectionType("nonmapped_type").system_type
 
     def test_system_key(self, db, connection_config, system):