def test_constant_initialize_creates_constant_iterator(self):
value = 1
c = pf.Constant(value)
self.assertEqual(len(c.parents), 1)
self.assertIsInstance(c.parents[0], pf.core.constant._ConstantIterator)
def test_broadcast_initialize_sets_count(self):
expected_count = pf.Constant(42)
broadcast = pf.flow.Broadcast(pf.Placeholder(out_type=int),
count=expected_count)
self.assertEqual(broadcast.count, expected_count)
def test_constant_apply_returns_value(self):
c = pf.Constant('asdf')
value = 3
x = [val for val in c.apply(value)]
self.assertEqual(len(x), 1)
self.assertEqual(x[0], value)
def setUp(self):
self.img_id = 'someID'
self.img_data = np.random.randn(10, 10, 3)
self.image = pf.Image(self.img_id, 'jpg', self.img_data)
self.mask = pf.transform.ColorMask(pf.Placeholder(out_type=pf.Image),
pf.Constant([None, None, None]))
def test_colormask_initialize_accepts_masks(self):
target_mask = [138, 222, 156]
mask = pf.transform.ColorMask(pf.Placeholder(out_type=pf.Image),
pf.Constant(target_mask))
self.assertEqual(len(mask.parents), 2)
self.assertIsInstance(mask.parents[1], pf.Constant)
self.assertEqual(next(mask.parents[1]), target_mask)
def test_broadcast_apply_correctly_broadcasts_values(self):
input_vals = [1, 2, 3, 4]
count = 2
expected_out = [1, 1, 2, 2, 3, 3, 4, 4]
broadcast = pf.flow.Broadcast(pf.Placeholder(out_type=int),
count=pf.Constant(42))
actual_out = []
for val in input_vals:
actual_out += [x for x in broadcast.apply(val, count)]
self.assertEqual(actual_out, expected_out)
def test_constant_initialize_sets_output_type(self):
value = 1
c = pf.Constant(value)
self.assertEqual(c._output_type, type(value))
def test_broadcast_initialize_defaults_id_to_broadcast(self):
broadcast = pf.flow.Broadcast(pf.Placeholder(out_type=int),
count=pf.Constant(42))
self.assertEqual(broadcast.id, 'broadcast')