Skip to content

Commit 8b07299

Browse files
committed
Add test
1 parent ee9ee1d commit 8b07299

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

ccdproc/tests/test_image_collection.py

+16
Original file line numberDiff line numberDiff line change
@@ -1100,3 +1100,19 @@ def test_filtered_collection_with_no_files(self, triage_setup):
11001100
ifc = ImageFileCollection(triage_setup.test_dir)
11011101

11021102
ifc_no_files = ifc.filter(object='really fake object')
1103+
1104+
@pytest.mark.parametrize('extensions', ('flubber', ['flubber']))
1105+
def test_user_specified_file_extensions(self, tmp_path, extensions):
1106+
# Test for #727, allowing user to specify fits
1107+
# extensions
1108+
ccd = CCDData(data=np.zeros([10, 10]), unit='adu')
1109+
num_files = 4
1110+
if len(extensions) == 1:
1111+
extension = extensions[0]
1112+
else:
1113+
extension = extensions
1114+
# Explicitly give the format for writing
1115+
_ = [ccd.write(tmp_path / f"ccd_{i}.{extension}", format='fits')
1116+
for i in range(num_files)]
1117+
ifc = ImageFileCollection(tmp_path, extensions=extensions)
1118+
assert len(ifc.summary) == num_files

0 commit comments

Comments
 (0)