fix tests

Author: vvkh

data_diff/info_tree.py

@@ -18,7 +18,7 @@ class SegmentInfo:
     rowcounts: Dict[int, int] = {}
     max_rows: int = None
 
-    def set_diff(self, schema: tuple[tuple[str, type]], diff: list[tuple[Any, ...]]):
+    def set_diff(self, diff: list[tuple[Any, ...]], schema: tuple[tuple[str, type]] | None = None):
         self.diff_schema = schema
         self.diff = diff
         self.diff_count = len(diff)

data_diff/joindiff_tables.py

@@ -206,7 +206,7 @@ def _diff_segments(
             assert len(a_cols) == len(b_cols)
             logger.debug("Querying for different rows")
             diff = db.query(diff_rows, list)
-            info_tree.info.set_diff(tuple(diff_rows.schema.items()), diff)
+            info_tree.info.set_diff(diff, schema=tuple(diff_rows.schema.items()))
             for is_xa, is_xb, *x in diff:
                 if is_xa and is_xb:
                     # Can't both be exclusive, meaning a pk is NULL