wip

xbrianh · Sep 10, 2024 · 0f9fdca · 0f9fdca
1 parent 7154856
commit 0f9fdca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_delta_log.py b/tests/test_delta_log.py
@@ -32,7 +32,7 @@ def test_schema(self):
             ],
             type="struct",
         )
-        dlog = xdlake.read_delta_log(LOGDIR)
+        dlog = delta_log.DeltaLog.with_location(LOGDIR)
         s = dlog.schema()
         self.assertEqual(s, expected)
 
@@ -54,7 +54,7 @@ def test_partition_columns(self):
         for expected_partition_columns in (list(), [f"{uuid4()}" for _ in range(3)]):
             with self.subTest(expected=expected_partition_columns):
                 tc = delta_log.DeltaLogEntry.commit_overwrite_table(expected_partition_columns, [], [])
-                dlog = xdlake.read_delta_log(LOGDIR)
+                dlog = delta_log.DeltaLog.with_location(LOGDIR)
                 dlog.entries[len(dlog.entries)] = tc
                 self.assertEqual(expected_partition_columns, dlog.partition_columns())