From af21bf86529ace14c151e595724dd626941284a5 Mon Sep 17 00:00:00 2001 From: The TensorFlow Datasets Authors Date: Mon, 20 Jan 2025 01:25:26 -0800 Subject: [PATCH] Make test more readable PiperOrigin-RevId: 717440673 --- .../core/dataset_builders/croissant_builder_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow_datasets/core/dataset_builders/croissant_builder_test.py b/tensorflow_datasets/core/dataset_builders/croissant_builder_test.py index e3722703cae..66bfbae17cb 100644 --- a/tensorflow_datasets/core/dataset_builders/croissant_builder_test.py +++ b/tensorflow_datasets/core/dataset_builders/croissant_builder_test.py @@ -245,7 +245,8 @@ def test_download_and_prepare(crs_builder, expected_entries, split_name): expected_entries = [ entry for entry in expected_entries if entry["split"] == split_name ] - assert len(data_source) == len(expected_entries) == 1 + assert len(data_source) == 1 + assert len(expected_entries) == 1 for entry, expected_entry in zip(data_source, expected_entries): assert entry["index"] == expected_entry["index"] assert entry["text"].decode() == expected_entry["text"]