Make test more readable

PiperOrigin-RevId: 717440673
tensorflow · Jan 20, 2025 · ded5e83 · ded5e83
1 parent cb1ae5a
commit ded5e83
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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(data_source) == 1
   for entry, expected_entry in zip(data_source, expected_entries):
     assert entry["index"] == expected_entry["index"]
     assert entry["text"].decode() == expected_entry["text"]