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')