Update test_dm_reader.py

pycroscopy · Sep 10, 2023 · 63b1759 · 63b1759
1 parent d73f20a
commit 63b1759
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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')