Fix types; Fix test for presto

Author: erezsh

data_diff/table_segment.py

@@ -243,13 +243,13 @@ def count_and_checksum(self) -> Tuple[int, int]:
             assert checksum, (count, checksum)
         return count or 0, int(checksum) if count else None
 
-    def query_key_range(self) -> Tuple[int, int]:
+    def query_key_range(self) -> Tuple[tuple, tuple]:
         """Query database for minimum and maximum key. This is used for setting the initial bounds."""
         # Normalizes the result (needed for UUIDs) after the min/max computation
         select = self.make_select().select(
             ApplyFuncAndNormalizeAsString(this[k], f) for k in self.key_columns for f in (min_, max_)
         )
-        result = self.database.query(select, tuple)
+        result = tuple(self.database.query(select, tuple))
 
         if any(i is None for i in result):
             raise ValueError("Table appears to be empty")

tests/test_diff_tables.py

@@ -880,13 +880,14 @@ class TestCompoundKeyAlphanum(DiffTestCase):
     def setUp(self):
         super().setUp()
 
-        diffs = [(uuid.uuid1(i), str(i), str(i)) for i in range(100)]
+        rows = [(uuid.uuid1(i), i, str(i)) for i in range(100)]
+        rows2 = list(rows)
+        x = rows2[9]
+        rows2[9] = (x[0], 9000, x[2])
         self.connection.query(
             [
-                self.src_table.insert_rows(diffs),
-                commit,
-                self.dst_table.insert_expr(self.src_table),
-                code("UPDATE {tbl} SET id2 = 9000 WHERE id2 = 9", tbl=self.dst_table),
+                self.src_table.insert_rows(rows),
+                self.dst_table.insert_rows(rows2),
                 commit,
             ]
         )