diff --git a/tests/readers/microscopy/em/tem/test_dm_reader.py b/tests/readers/microscopy/em/tem/test_dm_reader.py index 5cea6ff..7417852 100644 --- a/tests/readers/microscopy/em/tem/test_dm_reader.py +++ b/tests/readers/microscopy/em/tem/test_dm_reader.py @@ -25,7 +25,7 @@ def test_load_dm3_file(self): file_name = wget.download(data_path + '/DMReader_EELS_STO.dm3') reader = SciFiReaders.DMReader(file_name) datasets = reader.read() - dataset = datasets[0] + dataset = datasets['Channel_000'] self.assertEqual(dataset.title, '01-EELS Acquire_STO') self.assertEqual(dataset.source, 'SciFiReaders.DMReader') self.assertEqual(dataset[200].compute(), 135727.0) @@ -40,7 +40,7 @@ def test_load_dm4_file(self): file_name = wget.download(data_path + '/DMReader_EELS_STO.dm4') reader = SciFiReaders.DMReader(file_name, verbose=True) datasets = reader.read() - dataset = datasets[0] + dataset = datasets['Channel_000'] self.assertEqual(dataset.title, 'EELS_STO') self.assertEqual(dataset.source, 'SciFiReaders.DMReader') @@ -70,7 +70,7 @@ def test_load_dm3_spectrum_image(self): reader = SciFiReaders.DMReader(file_name) datasets = reader.read() - dataset = datasets[0] + dataset = datasets['Channel_000'] self.assertEqual(dataset.title, '06-EELS Spectrum Image') self.assertEqual(dataset.source, 'SciFiReaders.DMReader') @@ -90,7 +90,7 @@ def test_load_dm3_image(self): reader = SciFiReaders.DMReader(file_name) datasets = reader.read() - dataset = datasets[0] + dataset = datasets['Channel_000'] self.assertEqual(dataset.title, '06-SI Survey Image') self.assertEqual(dataset.source, 'SciFiReaders.DMReader')