1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97
|
# coding: utf-8
import unittest
from nxtomomill.io import utils
from nxtomomill.io.framegroup import FrameGroup
class TestConvertStrToTuple(unittest.TestCase):
"""
test convert_str_to_tuple function
"""
def testStr1(self):
self.assertEqual(utils.convert_str_to_tuple("toto, tata"), ("toto", "tata"))
def testStr2(self):
self.assertEqual(
utils.convert_str_to_tuple("'toto', \"tata\""), ("toto", "tata")
)
def testStr3(self):
self.assertEqual(utils.convert_str_to_tuple("test"), ("test",))
def testStr4(self):
self.assertEqual(
utils.convert_str_to_tuple("(this is a test)"), ("this is a test",)
)
def testStr5(self):
self.assertEqual(
utils.convert_str_to_tuple("(this is a test, 'and another one')"),
("this is a test", "and another one"),
)
class TestIsUrlPath(unittest.TestCase):
"""test the is_url function"""
def test_invalid_url_1(self):
self.assertFalse(utils.is_url_path("/toto/tata"))
def test_invalid_url_2(self):
self.assertFalse(utils.is_url_path("tata"))
def test_valid_url_1(self):
self.assertTrue(utils.is_url_path("silx:///data/image.h5?path=/scan_0/data"))
def test_valid_url_2(self):
self.assertTrue(utils.is_url_path("silx:///data/image.edf"))
def test_valid_url_3(self):
self.assertTrue(utils.is_url_path("silx://image.h5"))
def test_valid_url_4(self):
self.assertTrue(
utils.is_url_path("silx:///data/image.h5?path=/scan_0/data&slice=1,5")
)
class TestConvertStrToFrameGrp(unittest.TestCase):
"""
test the `convert_str_to_frame_grp` function
"""
def test_valid_1(self):
dg1 = FrameGroup(
frame_type="dark", url="silx:///file.h5?data_path=/dark", copy=True
)
dg2 = FrameGroup(
frame_type="flat", url="silx:///file.h5?data_path=/flat", copy=True
)
dg3 = FrameGroup(
frame_type="projection",
url="silx:///file.h5?data_path=/projection1",
copy=True,
)
dg4 = FrameGroup(
frame_type="alignment",
url="silx:///file.h5?data_path=/projection2",
copy=True,
)
my_str = "({dg1}, {dg2}, {dg3}, {dg4}, )".format(
dg1=dg1, dg2=dg2, dg3=dg3, dg4=dg4
)
frame_grps = utils.convert_str_to_frame_grp(my_str)
self.assertTrue(isinstance(frame_grps, tuple))
self.assertTrue(len(frame_grps) == 4)
for o_fg, i_fg in zip((dg1, dg2, dg3, dg4), frame_grps):
self.assertEqual(str(o_fg), str(i_fg))
def test_valid_2(self):
pass
def test_invalid_1(self):
pass
|