def setUp(self):
self.folder_in = tempfile.mkdtemp()
for i in range(10):
DicommlCase(**sample_case_config(
caseid='train-case-{i}'.format(i=i), n_images=10)).save(
os.path.join(self.folder_in, 'train'))
for i in range(4):
DicommlCase(**sample_case_config(
caseid='eval-case-{i}'.format(i=i), n_images=10)).save(
os.path.join(self.folder_in, 'eval'))
class TestExpandTransforms(unittest.TestCase):
def test_split(self):
case = DicommlCase(**sample_case_config(n_images=10))
cases_new = expand.Split(n_images=2)([case])
self.assertEqual(len(cases_new), 5)
for case in cases_new:
self.assertIsInstance(case, DicommlCase)
class TestArrayTransforms(unittest.TestCase):
def test_shift(self):
case = DicommlCase(**sample_case_config())
case_new = transforms.Shift(x_shift=2, y_shift=34)([case])[0]
self.assertEqual(case.export()['images'].shape,
case_new.export()['images'].shape)
from dicomml.cases.case import DicommlCase
from dicomml.transforms import array as transforms
from dicomml.transforms import expand
from tests import sample_case_config
class TestArrayTransforms(unittest.TestCase):
def test_shift(self):
case = DicommlCase(**sample_case_config())
case_new = transforms.Shift(x_shift=2, y_shift=34)([case])[0]
self.assertEqual(case.export()['images'].shape,
case_new.export()['images'].shape)
def test_mirror(self):
case = DicommlCase(**sample_case_config())
case_new = transforms.Mirror()([case])[0]
self.assertEqual(case.export()['images'].shape,
case_new.export()['images'].shape)
def test_cut(self):
case = DicommlCase(**sample_case_config())
case_new = transforms.Cut(x_range=[30, 90], y_range=[20,
100])([case])[0]
self.assertEqual((1, case.export()['images'].shape[1], 60, 80),
case_new.export()['images'].shape)
def test_pad(self):
case = DicommlCase(**sample_case_config())
case_new = transforms.Pad(target_shape=[200, 150])([case])[0]
self.assertEqual((1, case.export()['images'].shape[1], 200, 150),
class TestDicommlCase(unittest.TestCase):
def test_create(self):
case = DicommlCase(**sample_case_config())
self.assertIsInstance(case, DicommlCase)
import unittest
import tempfile
from dicomml.cases.case import DicommlCase
from tests import sample_case_config
class TestDicommlCase(unittest.TestCase):
def test_create(self):
case = DicommlCase(**sample_case_config())
self.assertIsInstance(case, DicommlCase)
def test_save_load(self):
case = DicommlCase(**sample_case_config())
with tempfile.TemporaryDirectory() as temp_folder:
zipfile = case.save(path=temp_folder)
case_loaded = DicommlCase.load(zipfile)
self.assertEqual(case, case_loaded)
def test_export(self):
case = DicommlCase(**sample_case_config())
exports = case.export()
self.assertCountEqual(
list(exports.keys()),
['images', 'truth'])
self.assertEqual(exports['images'].shape, (1, 10, 120, 120))
def setUp(self):
self.folder_in = tempfile.mkdtemp()
self.folder_out = tempfile.mkdtemp()
for i in range(4):
DicommlCase(**sample_case_config(caseid='case-{i}'.format(i=i),
n_images=10)).save(self.folder_in)