def test_invalid_input(self):
with self.assertRaisesRegexp(ValueError,
'are not contained in the set'):
ops.pad(self.original_lt, {
'foo': (1, 1),
'channel': ([], ['alpha'])
})
def test(self):
pad_lt = ops.pad(self.original_lt, {'x': (1, 1),
'channel': ([], ['alpha'])})
golden_op = tf.pad(self.original_lt.tensor, [[1, 1], [0, 1], [0, 0],
[0, 0]])
golden_axes = [('x', self.x_size + 2),
('channel', ['red', 'green', 'blue', 'alpha']), self.a2,
self.a3]
golden_lt = core.LabeledTensor(golden_op, golden_axes)
self.assertLabeledTensorsEqual(pad_lt, golden_lt)
def test_name(self):
pad_lt = ops.pad(self.original_lt, {
'x': (1, 1),
'channel': ([], ['alpha'])
})
self.assertIn('lt_pad', pad_lt.name)
def test_invalid_input(self):
with self.assertRaisesRegexp(ValueError, 'are not contained in the set'):
ops.pad(self.original_lt, {'foo': (1, 1), 'channel': ([], ['alpha'])})
def test_name(self):
pad_lt = ops.pad(self.original_lt,
{'x': (1, 1),
'channel': ([], ['alpha'])})
self.assertIn('lt_pad', pad_lt.name)
