Silence some mypy warnings

Author: mhauru

mypy.ini

@@ -10,3 +10,4 @@ warn_return_any = True
 show_error_codes = True
 warn_unused_ignores = True
 ignore_missing_imports = True
+plugins = pydantic.mypy

sqlsynthgen/utils.py

@@ -70,7 +70,7 @@ def create_db_engine(
 
         @event.listens_for(event_engine, "connect", insert=True)
         def connect(dbapi_connection: Any, _: Any) -> None:
-            set_search_path(dbapi_connection, schema_name)  # type: ignore
+            set_search_path(dbapi_connection, schema_name)
 
     return engine
 

tests/test_remove.py

@@ -31,7 +31,11 @@ def test_remove_db_data(
     @patch("sqlsynthgen.remove.get_settings")
     def test_remove_db_data_raises(self, mock_get: MagicMock) -> None:
         """Check that remove_db_data raises if dst DSN is missing."""
-        mock_get.return_value = Settings(dst_dsn=None, _env_file=None)
+        mock_get.return_value = Settings(
+            dst_dsn=None,
+            # The mypy ignore can be removed once we upgrade to pydantic 2.
+            _env_file=None,  # type: ignore[call-arg]
+        )
         with self.assertRaises(AssertionError) as context_manager:
             remove_db_data(example_orm, remove_ssg)
         self.assertEqual(
@@ -56,7 +60,11 @@ def test_remove_db_vocab(
     @patch("sqlsynthgen.remove.get_settings")
     def test_remove_db_vocab_raises(self, mock_get: MagicMock) -> None:
         """Check that remove_db_vocab raises if dst DSN is missing."""
-        mock_get.return_value = Settings(dst_dsn=None, _env_file=None)
+        mock_get.return_value = Settings(
+            dst_dsn=None,
+            # The mypy ignore can be removed once we upgrade to pydantic 2.
+            _env_file=None,  # type: ignore[call-arg]
+        )
         with self.assertRaises(AssertionError) as context_manager:
             remove_db_vocab(example_orm, remove_ssg)
         self.assertEqual(
@@ -75,7 +83,11 @@ def test_remove_tables(self, mock_engine: MagicMock, _: MagicMock) -> None:
     @patch("sqlsynthgen.remove.get_settings")
     def test_remove_db_tables_raises(self, mock_get: MagicMock) -> None:
         """Check that remove_db_tables raises if dst DSN is missing."""
-        mock_get.return_value = Settings(dst_dsn=None, _env_file=None)
+        mock_get.return_value = Settings(
+            dst_dsn=None,
+            # The mypy ignore can be removed once we upgrade to pydantic 2.
+            _env_file=None,  # type: ignore[call-arg]
+        )
         with self.assertRaises(AssertionError) as context_manager:
             remove_db_tables(example_orm)
         self.assertEqual(

tests/test_settings.py

@@ -12,7 +12,8 @@ def test_minimal_settings(self) -> None:
         """Test the minimal settings."""
         settings = Settings(
             # To stop any local .env files influencing the test
-            _env_file=None,
+            # The mypy ignore can be removed once we upgrade to pydantic 2.
+            _env_file=None,  # type: ignore[call-arg]
         )
         self.assertIsNone(settings.src_dsn)
         self.assertIsNone(settings.src_schema)
@@ -28,7 +29,8 @@ def test_maximal_settings(self) -> None:
             dst_dsn="postgresql://user:password@host:port/db_name?sslmode=require",
             dst_schema="src_schema",
             # To stop any local .env files influencing the test
-            _env_file=None,
+            # The mypy ignore can be removed once we upgrade to pydantic 2.
+            _env_file=None,  # type: ignore[call-arg]
         )
 
     def test_validation(self) -> None:

tests/utils.py

@@ -21,7 +21,8 @@ def get_test_settings() -> settings.Settings:
         src_dsn="postgresql://suser:spassword@shost:5432/sdbname",
         dst_dsn="postgresql://duser:dpassword@dhost:5432/ddbname",
         # To stop any local .env files influencing the test
-        _env_file=None,
+        # The mypy ignore can be removed once we upgrade to pydantic 2.
+        _env_file=None,  # type: ignore[call-arg]
     )