From 0a1ad2e764c42f819b3b1f9207f673eeeaee52f6 Mon Sep 17 00:00:00 2001 From: Will Jones Date: Wed, 30 Aug 2023 14:00:38 -0700 Subject: [PATCH] update test for roundtrip --- python/python/tests/test_tf.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/python/python/tests/test_tf.py b/python/python/tests/test_tf.py index 59978a27d8..acc0312df3 100644 --- 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