update test for roundtrip

lancedb · Aug 30, 2023 · 0a1ad2e · 0a1ad2e
1 parent d47144e
commit 0a1ad2e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/python/python/tests/test_tf.py b/python/python/tests/test_tf.py
@@ -266,6 +266,12 @@ def test_tfrecord_parsing(tmp_path):
 
     assert table == expected_data
 
+    # Can roundtrip to lance
+    dataset_uri = tmp_path / "dataset"
+    dataset = lance.write_dataset(table, dataset_uri)
+    assert dataset.schema == expected_data.schema
+    assert dataset.to_table() == expected_data
+
 
 def test_tfrecord_parsing_nulls(tmp_path):
     # Make sure we don't trip up on missing values