re-introduce count check

Energinet-DataHub · Nov 29, 2024 · fc37189 · fc37189
1 parent 2185aae
commit fc37189
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/source/databricks/calculation_engine/tests/helpers/data_frame_utils.py b/source/databricks/calculation_engine/tests/helpers/data_frame_utils.py
@@ -155,6 +155,20 @@ def assert_dataframe_and_schema(
         expected.subtract(actual).show(3000, False)
         raise
 
+    try:
+        assert actual.count() == expected.count()
+    except AssertionError:
+
+        if (
+            not feature_tests_configuration.show_columns_when_actual_and_expected_are_equal
+        ):
+            actual, expected = drop_columns_if_the_same(actual, expected)
+
+        print(
+            f"NUMBER OF ROWS MISMATCH: Actual: {actual.count()}, Expected: {expected.count()}"
+        )
+        raise
+
 
 def drop_columns_if_the_same(df1: DataFrame, df2: DataFrame) -> (DataFrame, DataFrame):
     column_names = df1.columns