Skip to content
Snippets Groups Projects
Commit 735ca41a authored by Christopher Randolph Rhodes's avatar Christopher Randolph Rhodes
Browse files

Added support for RGB PNG

parent 9fe25e61
No related branches found
No related tags found
No related merge requests found
......@@ -12,6 +12,16 @@ czifile = {
'z': 1,
}
filename = 'test_img.png'
pngfile = {
'filename': filename,
'path': root / filename,
'w': 64,
'h': 128,
'c': 3,
'z': 1
}
filename = 'zmask-test-stack.tif'
tifffile = {
'filename': filename,
......
......@@ -2,8 +2,8 @@ import unittest
import numpy as np
from conf.testing import czifile, output_path, tifffile
from model_server.accessors import CziImageFileAccessor, DataShapeError, generate_file_accessor, InMemoryDataAccessor, write_accessor_data_to_file, TifSingleSeriesFileAccessor
from conf.testing import czifile, output_path, pngfile, tifffile
from model_server.accessors import CziImageFileAccessor, DataShapeError, generate_file_accessor, InMemoryDataAccessor, PngFileAccessor, write_accessor_data_to_file, TifSingleSeriesFileAccessor
class TestCziImageFileAccess(unittest.TestCase):
......@@ -98,3 +98,9 @@ class TestCziImageFileAccess(unittest.TestCase):
se = fh.series[0]
fh_shape_dict = {se.axes[i]: se.shape[i] for i in range(0, len(se.shape))}
self.assertEqual(fh_shape_dict, acc.shape_dict, 'Axes are not preserved in TIF output')
def test_read_rgb_png(self):
acc = PngFileAccessor(pngfile['path'])
self.assertEqual(acc.hw, (pngfile['h'], pngfile['w']))
self.assertEqual(acc.chroma, 3)
self.assertEqual(acc.nz, 1)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment