From 4a996ce8f9c109ed03be71d221524d7c276499db Mon Sep 17 00:00:00 2001 From: Valentyn Tymofieiev Date: Wed, 13 Sep 2023 14:22:54 -0700 Subject: [PATCH] Remove the test that looks for exception: pyarrow==14 now handles this conversion. --- sdks/python/apache_beam/io/parquetio_test.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/sdks/python/apache_beam/io/parquetio_test.py b/sdks/python/apache_beam/io/parquetio_test.py index df018a3a776f..bafe10e560b3 100644 --- a/sdks/python/apache_beam/io/parquetio_test.py +++ b/sdks/python/apache_beam/io/parquetio_test.py @@ -333,18 +333,6 @@ def test_write_batched_display_data(self): ] hc.assert_that(dd.items, hc.contains_inanyorder(*expected_items)) - def test_sink_transform_int96(self): - with tempfile.NamedTemporaryFile() as dst: - path = dst.name - # pylint: disable=c-extension-no-member - with self.assertRaises(pl.ArrowInvalid): - # Should throw an error "ArrowInvalid: Casting from timestamp[ns] to - # timestamp[us] would lose data" - with TestPipeline() as p: - _ = p \ - | Create(self.RECORDS) \ - | WriteToParquet( - path, self.SCHEMA96, num_shards=1, shard_name_template='') def test_sink_transform(self): with TemporaryDirectory() as tmp_dirname: