|
|
@ -0,0 +1,42 @@ |
|
|
|
from nd2reader.model.image import Image |
|
|
|
import numpy as np |
|
|
|
import unittest |
|
|
|
|
|
|
|
|
|
|
|
class ImageTests(unittest.TestCase): |
|
|
|
""" |
|
|
|
Basically just tests that the Image API works and that Images act as Numpy arrays. There's very little going on |
|
|
|
here other than simply storing data. |
|
|
|
|
|
|
|
""" |
|
|
|
def setUp(self): |
|
|
|
array = np.array([[0, 1, 254], |
|
|
|
[45, 12, 9], |
|
|
|
[12, 12, 99]]) |
|
|
|
self.image = Image(array) |
|
|
|
self.image.add_params(1200.314, 17, 2, 'GFP', 1) |
|
|
|
|
|
|
|
def test_size(self): |
|
|
|
self.assertEqual(self.image.height, 3) |
|
|
|
self.assertEqual(self.image.width, 3) |
|
|
|
|
|
|
|
def test_timestamp(self): |
|
|
|
self.assertEqual(self.image.timestamp, 1.200314) |
|
|
|
|
|
|
|
def test_frame_number(self): |
|
|
|
self.assertEqual(self.image.frame_number, 17) |
|
|
|
|
|
|
|
def test_fov(self): |
|
|
|
self.assertEqual(self.image.field_of_view, 2) |
|
|
|
|
|
|
|
def test_channel(self): |
|
|
|
self.assertEqual(self.image.channel, 'GFP') |
|
|
|
|
|
|
|
def test_z_level(self): |
|
|
|
self.assertEqual(self.image.z_level, 1) |
|
|
|
|
|
|
|
def test_slice(self): |
|
|
|
subimage = self.image[:2, :2] |
|
|
|
expected = np.array([[0, 1], |
|
|
|
[45, 12]]) |
|
|
|
self.assertTrue(np.array_equal(subimage, expected)) |