Show column types in assert_subset_df

MobileTeleSystems · Dec 5, 2024 · 2865f26 · 2865f26
1 parent 2296982
commit 2865f26
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/util/assert_df.py b/tests/util/assert_df.py
@@ -66,5 +66,12 @@ def assert_subset_df(
         columns = [column.lower() for column in columns]
 
     for column in columns:  # noqa: WPS528
-        difference = ~small_pdf[column].isin(large_pdf[column])
-        assert not difference.all(), small_pdf[difference]
+        small_column = small_pdf[column]
+        large_column = large_pdf[column]
+        different_indices = ~small_column.isin(large_column)
+        assert not different_indices.all(), (
+            column,
+            small_column.dtype,
+            large_column.dtype,
+            small_pdf[different_indices],
+        )