nd2reader/tests/test_reader.py

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())