import unittest
|
|
from os import path
|
|
from zolfa.nd2reader.artificial import ArtificialND2
|
|
from zolfa.nd2reader.common import check_or_make_dir
|
|
from zolfa.nd2reader.parser import Parser
|
|
import urllib.request
|
|
|
|
|
|
class TestParser(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__))
|
|
check_or_make_dir(path.join(dir_path, "test_data/"))
|
|
self.test_file = path.join(dir_path, "test_data/test.nd2")
|
|
self.create_test_nd2()
|
|
|
|
def test_can_open_test_file(self):
|
|
self.create_test_nd2()
|
|
|
|
with open(self.test_file, "rb") as fh:
|
|
parser = Parser(fh)
|
|
self.assertTrue(parser.supported)
|
|
|
|
def test_get_image(self):
|
|
stitched_path = "test_data/test_stitched.nd2"
|
|
if not path.isfile(stitched_path):
|
|
file_name, header = urllib.request.urlretrieve(
|
|
"https://downloads.openmicroscopy.org/images/ND2/karl/sample_image.nd2",
|
|
stitched_path,
|
|
)
|
|
with open(stitched_path, "rb") as fh:
|
|
parser = Parser(fh)
|
|
parser.get_image(0)
|