Skip to content

Commit 66477f7

Browse files
committed
Make current_records flag test more explicit
1 parent bcaa7b3 commit 66477f7

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

tests/test_dataset.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,9 +138,13 @@ def test_dataset_load_with_multi_nonpartition_filters_success(fixed_local_datase
138138

139139
def test_dataset_load_current_records_all_sources_success(dataset_with_runs_location):
140140
timdex_dataset = TIMDEXDataset(dataset_with_runs_location)
141-
timdex_dataset.load(current_records=True)
142141

143-
# 14 total parquet files, only 12 related to current runs
142+
# 16 total parquet files, with current_records=False we get them all
143+
timdex_dataset.load(current_records=False)
144+
assert len(timdex_dataset.dataset.files) == 16
145+
146+
# 16 total parquet files, with current_records=True we only get 12 for current runs
147+
timdex_dataset.load(current_records=True)
144148
assert len(timdex_dataset.dataset.files) == 12
145149

146150

0 commit comments

Comments
 (0)