def test_image_gradient(self): img = Image(np.array([[0, 1, 0], [3, 4, 5.], [6, 7, 8]])) op_img = gradient(img) self.assertEqual( op_img, img.clone(np.array([[3.162278, 3.162278], [3.162278, 3.162278]]), group='Gradient'))
def test_image_transform(self): img = Image(np.random.rand(10, 10)) op_img = transform(img, operator=lambda x: x * 2) self.assertEqual(op_img, img.clone(img.data * 2, group='Transform'))
def test_image_threshold(self): img = Image(np.array([[0, 1, 0], [3, 4, 5.]])) op_img = threshold(img) self.assertEqual( op_img, img.clone(np.array([[0, 1, 0], [1, 1, 1]]), group='Threshold'))
def test_operation_element(self): img = Image(np.random.rand(10, 10)) op_img = operation(img, op=lambda x, k: x.clone(x.data * 2)) self.assertEqual(op_img, img.clone(img.data * 2, group='Operation'))
def test_image_gradient(self): img = Image(np.array([[0, 1, 0], [3, 4, 5.], [6, 7, 8]])) op_img = gradient(img) self.assertEqual(op_img, img.clone(np.array([[3.162278, 3.162278], [3.162278, 3.162278]]), group='Gradient'))
def test_image_threshold(self): img = Image(np.array([[0, 1, 0], [3, 4, 5.]])) op_img = threshold(img) self.assertEqual(op_img, img.clone(np.array([[0, 1, 0], [1, 1, 1]]), group='Threshold'))
def test_image_transform(self): img = Image(np.random.rand(10, 10)) op_img = transform(img, operator=lambda x: x*2) self.assertEqual(op_img, img.clone(img.data*2, group='Transform'))
def test_operation_element(self): img = Image(np.random.rand(10, 10)) op_img = operation(img, op=lambda x, k: x.clone(x.data*2)) self.assertEqual(op_img, img.clone(img.data*2, group='Operation'))