def test_getting_value_for_keyword(self, triage_setup): ic = ImageFileCollection(triage_setup.test_dir, keywords=['imagetyp']) # Does it fail if the keyword is not in the summary? with pytest.raises(ValueError): ic.values('filter') # If I ask for unique values do I get them? values = ic.values('imagetyp', unique=True) assert values == list(set(ic.summary['imagetyp'])) assert len(values) < len(ic.summary['imagetyp']) # Does the list of non-unique values match the raw column? values = ic.values('imagetyp', unique=False) assert values == list(ic.summary['imagetyp']) # Does unique actually default to false? values2 = ic.values('imagetyp') assert values == values2
def test_filter_files(self, triage_setup): img_collection = ImageFileCollection( location=triage_setup.test_dir, keywords=['imagetyp', 'filter']) assert len(img_collection.files_filtered( imagetyp='bias')) == triage_setup.n_test['bias'] assert len(img_collection.files) == triage_setup.n_test['files'] assert ('filter' in img_collection.keywords) assert ('flying monkeys' not in img_collection.keywords) assert len(img_collection.values('imagetyp', unique=True)) == 2