def test_extract_simple_patches_2d_color_float(self):
image = np.arange(0.0, 4.8, 0.1, dtype='float').reshape((4, 4, 3))
actual = extract_simple_patches_2d(image, (
2,
2,
))
self.assertEqual((4, 2, 2, 3), actual.shape)
self.assertEqual(image.dtype, actual.dtype)
assert_array_almost_equal(
np.array(
[0.0, 0.1, 0.2, 0.3, 0.4, 0.5, 1.2, 1.3, 1.4, 1.5, 1.6, 1.7]),
actual[0].flatten())
assert_array_almost_equal(
np.array(
[0.6, 0.7, 0.8, 0.9, 1.0, 1.1, 1.8, 1.9, 2.0, 2.1, 2.2, 2.3]),
actual[1].flatten())
assert_array_almost_equal(
np.array(
[2.4, 2.5, 2.6, 2.7, 2.8, 2.9, 3.6, 3.7, 3.8, 3.9, 4.0, 4.1]),
actual[2].flatten())
assert_array_almost_equal(
np.array(
[3.0, 3.1, 3.2, 3.3, 3.4, 3.5, 4.2, 4.3, 4.4, 4.5, 4.6, 4.7]),
actual[3].flatten())
def test_extract_simple_patches_2d_gray(self):
image = np.arange(16).reshape((4, 4))
actual = extract_simple_patches_2d(image, (
2,
2,
))
self.assertEqual((4, 2, 2), actual.shape)
self.assertTrue((np.array([0, 1, 4, 5]) == actual[0].flatten()).all())
self.assertTrue((np.array([2, 3, 6, 7]) == actual[1].flatten()).all())
self.assertTrue((np.array([8, 9, 12,
13]) == actual[2].flatten()).all())
self.assertTrue((np.array([10, 11, 14,
15]) == actual[3].flatten()).all())
def test_extract_simple_patches_2d_gray_float(self):
image = np.arange(0.0, 1.6, 0.1, dtype='float').reshape((4, 4))
actual = extract_simple_patches_2d(image, (
2,
2,
))
self.assertEqual((4, 2, 2), actual.shape)
self.assertEqual(image.dtype, actual.dtype)
assert_array_almost_equal(np.array([0.0, 0.1, 0.4, 0.5]),
actual[0].flatten())
assert_array_almost_equal(np.array([0.2, 0.3, 0.6, 0.7]),
actual[1].flatten())
assert_array_almost_equal(np.array([0.8, 0.9, 1.2, 1.3]),
actual[2].flatten())
assert_array_almost_equal(np.array([1.0, 1.1, 1.4, 1.5]),
actual[3].flatten())
def test_extract_simple_patches_2d_color(self):
image = np.arange(48).reshape((4, 4, 3))
actual = extract_simple_patches_2d(image, (
2,
2,
))
self.assertEqual((4, 2, 2, 3), actual.shape)
self.assertTrue((np.array([0, 1, 2, 3, 4, 5, 12, 13, 14, 15, 16,
17]) == actual[0].flatten()).all())
self.assertTrue((np.array([6, 7, 8, 9, 10, 11, 18, 19, 20, 21, 22,
23]) == actual[1].flatten()).all())
self.assertTrue(
(np.array([24, 25, 26, 27, 28, 29, 36, 37, 38, 39, 40,
41]) == actual[2].flatten()).all())
self.assertTrue(
(np.array([30, 31, 32, 33, 34, 35, 42, 43, 44, 45, 46,
47]) == actual[3].flatten()).all())