Merge pull request #52 from yassun7010/fix_never_case

fix: never case.
yassun7010 · Jan 29, 2024 · bbf84d8 · bbf84d8
2 parents 8dc8ef6 + 37f02bf
commit bbf84d8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/turu-snowflake/src/turu/snowflake/mock_async_connection.py b/turu-snowflake/src/turu/snowflake/mock_async_connection.py
@@ -89,7 +89,11 @@ def inject_response(  # type: ignore[override]
         ],
         response: Union[Sequence[Any], Any, Exception] = None,
     ) -> Self:
-        if row_type is not None and isinstance(response, PandasDataFlame):
+        if (
+            row_type is not None
+            and PandasDataFlame is not Never
+            and isinstance(response, PandasDataFlame)
+        ):
             response = (response,)
 
         self._turu_mock_store.inject_response(

diff --git a/turu-snowflake/src/turu/snowflake/mock_connection.py b/turu-snowflake/src/turu/snowflake/mock_connection.py
@@ -87,7 +87,11 @@ def inject_response(  # type: ignore[override]
         ],
         response: Union[Sequence[Any], Any, Exception] = None,
     ) -> Self:
-        if row_type is not None and isinstance(response, PandasDataFlame):
+        if (
+            row_type is not None
+            and PandasDataFlame is not Never
+            and isinstance(response, PandasDataFlame)
+        ):
             response = (response,)
 
         self._turu_mock_store.inject_response(