From 63b1759584820976da810447a48e102072e49d22 Mon Sep 17 00:00:00 2001 From: Gerd Duscher <50049264+gduscher@users.noreply.github.com> Date: Mon, 11 Sep 2023 01:19:14 +0200 Subject: [PATCH] Update test_dm_reader.py --- tests/readers/microscopy/em/tem/test_dm_reader.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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')