fix mypy errors

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>

tests/dataframe/any_all_test.py

@@ -22,7 +22,7 @@ def test_reductions(
     df = bool_dataframe_1(library)
     ns = df.__dataframe_namespace__()
     result = getattr(df, reduction)()
-    compare_dataframe_with_reference(result, expected_data, dtype=ns.Bool)
+    compare_dataframe_with_reference(result, expected_data, dtype=ns.Bool)  # type: ignore[arg-type]
 
 
 def test_any(library: str) -> None:

tests/dataframe/comparisons_test.py

@@ -35,7 +35,7 @@ def test_comparisons_with_scalar(
     other = 2
     result = getattr(df, comparison)(other)
     expected_ns_dtype = getattr(ns, expected_dtype)
-    compare_dataframe_with_reference(result, expected_data, dtype=expected_ns_dtype)
+    compare_dataframe_with_reference(result, expected_data, dtype=expected_ns_dtype)  # type: ignore[arg-type]
 
 
 @pytest.mark.parametrize(
@@ -55,4 +55,4 @@ def test_rcomparisons_with_scalar(
     ns = df.__dataframe_namespace__()
     other = 2
     result = getattr(df, comparison)(other)
-    compare_dataframe_with_reference(result, expected_data, dtype=ns.Int64)
+    compare_dataframe_with_reference(result, expected_data, dtype=ns.Int64)  # type: ignore[arg-type]

tests/dataframe/fill_nan_test.py

@@ -39,6 +39,6 @@ def test_fill_nan_with_null(library: str) -> None:
     result = n_nans.col("a").persist().get_value(0).scalar
     if library == "pandas-numpy":
         # null is nan for pandas-numpy
-        assert result == 1  # type: ignore[index]
+        assert result == 1
     else:
-        assert result == 0  # type: ignore[index]
+        assert result == 0

tests/dataframe/join_test.py

@@ -20,7 +20,7 @@ def test_join_left(library: str) -> None:
         "b": ns.Int64,
         "c": ns.Int64 if library in ["pandas-nullable", "polars-lazy"] else ns.Float64,
     }
-    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)
+    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)  # type: ignore[arg-type]
 
 
 def test_join_overlapping_names(library: str) -> None:
@@ -60,7 +60,7 @@ def test_join_outer(library: str) -> None:  # pragma: no cover
         "b": ns.Int64 if library in ["pandas-nullable", "polars-lazy"] else ns.Float64,
         "c": ns.Int64 if library in ["pandas-nullable", "polars-lazy"] else ns.Float64,
     }
-    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)
+    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)  # type: ignore[arg-type]
 
 
 def test_join_two_keys(library: str) -> None:
@@ -74,7 +74,7 @@ def test_join_two_keys(library: str) -> None:
         "b": ns.Int64,
         "c": ns.Int64 if library in ["pandas-nullable", "polars-lazy"] else ns.Float64,
     }
-    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)
+    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)  # type: ignore[arg-type]
 
 
 def test_join_invalid(library: str) -> None:

tests/groupby/aggregate_test.py

@@ -53,7 +53,7 @@ def test_aggregate(library: str) -> None:
     }
     if library == "polars-lazy":
         result = result.cast({"b_count": ns.Int64()})
-    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)
+    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)  # type: ignore[arg-type]
 
 
 def test_aggregate_only_size(library: str) -> None:
@@ -108,4 +108,4 @@ def test_aggregate_no_size(library: str) -> None:
         "b_std": ns.Float64,
         "b_var": ns.Float64,
     }
-    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)
+    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)  # type: ignore[arg-type]

tests/groupby/groupby_any_all_test.py

@@ -36,7 +36,7 @@ def test_groupby_boolean(
         result = result.cast({"key": ns.Int64()})
     expected = {"key": [1, 2], "b": expected_b, "c": expected_c}
     expected_dtype = {"key": ns.Int64, "b": ns.Bool, "c": ns.Bool}
-    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)
+    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)  # type: ignore[arg-type]
 
 
 def test_group_by_invalid_any_all(library: str) -> None:

tests/groupby/numeric_test.py

@@ -46,4 +46,4 @@ def test_group_by_numeric(
     ):  # pragma: no cover
         # upstream bug
         result = result.cast({"key": ns.Int64()})
-    compare_dataframe_with_reference(result, expected, dtype=expected_ns_dtype)
+    compare_dataframe_with_reference(result, expected, dtype=expected_ns_dtype)  # type: ignore[arg-type]

tests/integration/persistedness_test.py

@@ -71,7 +71,7 @@ def test_cross_df_propagation(library: str) -> None:
         "b": ns.Int64,
         "c": ns.Int64 if library in ["pandas-nullable", "polars-lazy"] else ns.Float64,
     }
-    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)
+    compare_dataframe_with_reference(result, expected, dtype=expected_dtype)  # type: ignore[arg-type]
 
 
 def test_multiple_propagations(library: str) -> None: