You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
1.3 KiB
31 lines
1.3 KiB
import unittest
|
|
import numpy as np
|
|
|
|
from nd2reader.artificial import ArtificialND2
|
|
from nd2reader.exceptions import EmptyFileError
|
|
from nd2reader.reader import ND2Reader
|
|
|
|
|
|
class TestReader(unittest.TestCase):
|
|
def test_extension(self):
|
|
self.assertTrue('nd2' in ND2Reader.class_exts())
|
|
|
|
def test_init_and_init_axes(self):
|
|
with ArtificialND2('test_data/test_nd2_reader.nd2') as artificial:
|
|
reader = ND2Reader('test_data/test_nd2_reader.nd2')
|
|
|
|
attributes = artificial.data['image_attributes']['SLxImageAttributes']
|
|
self.assertEqual(reader.metadata['width'], attributes['uiWidth'])
|
|
self.assertEqual(reader.metadata['height'], attributes['uiHeight'])
|
|
|
|
self.assertEqual(reader.metadata['width'], reader.sizes['x'])
|
|
self.assertEqual(reader.metadata['height'], reader.sizes['y'])
|
|
|
|
self.assertEqual(reader._dtype, np.float64)
|
|
self.assertEqual(reader.iter_axes, ['t'])
|
|
|
|
def test_init_empty_file(self):
|
|
with ArtificialND2('test_data/empty.nd2', skip_blocks=['label_map_marker']):
|
|
with self.assertRaises(EmptyFileError) as exception:
|
|
ND2Reader('test_data/empty.nd2')
|
|
self.assertEqual(str(exception.exception), "No axes were found for this .nd2 file.")
|