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.
30 lines
1.0 KiB
30 lines
1.0 KiB
import unittest
|
|
from os import path
|
|
import numpy as np
|
|
|
|
from nd2reader.artificial import ArtificialND2
|
|
from nd2reader.reader import ND2Reader
|
|
|
|
|
|
class TestReader(unittest.TestCase):
|
|
def create_test_nd2(self):
|
|
with ArtificialND2(self.test_file) as artificial:
|
|
artificial.close()
|
|
|
|
def setUp(self):
|
|
dir_path = path.dirname(path.realpath(__file__))
|
|
self.test_file = path.join(dir_path, 'test_data/test.nd2')
|
|
|
|
def test_can_open_test_file(self):
|
|
self.create_test_nd2()
|
|
with ND2Reader(self.test_file) as reader:
|
|
self.assertEqual(reader.filename, self.test_file)
|
|
self.assertEqual(reader.pixel_type, np.float64)
|
|
self.assertEqual(reader.sizes['x'], 0)
|
|
self.assertEqual(reader.sizes['y'], 0)
|
|
self.assertEqual(reader.sizes['z'], 1)
|
|
self.assertEqual(reader.sizes['c'], 0)
|
|
self.assertEqual(reader.sizes['t'], 1)
|
|
|
|
def test_extension(self):
|
|
self.assertTrue('nd2' in ND2Reader.class_exts())
|