Chris@87: """Tests for the pad functions. Chris@87: Chris@87: """ Chris@87: from __future__ import division, absolute_import, print_function Chris@87: Chris@87: from numpy.testing import TestCase, run_module_suite, assert_array_equal Chris@87: from numpy.testing import assert_raises, assert_array_almost_equal Chris@87: import numpy as np Chris@87: from numpy.lib import pad Chris@87: Chris@87: Chris@87: class TestStatistic(TestCase): Chris@87: def test_check_mean_stat_length(self): Chris@87: a = np.arange(100).astype('f') Chris@87: a = pad(a, ((25, 20), ), 'mean', stat_length=((2, 3), )) Chris@87: b = np.array( Chris@87: [0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, Chris@87: 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, Chris@87: 0.5, 0.5, 0.5, 0.5, 0.5, Chris@87: Chris@87: 0., 1., 2., 3., 4., 5., 6., 7., 8., 9., Chris@87: 10., 11., 12., 13., 14., 15., 16., 17., 18., 19., Chris@87: 20., 21., 22., 23., 24., 25., 26., 27., 28., 29., Chris@87: 30., 31., 32., 33., 34., 35., 36., 37., 38., 39., Chris@87: 40., 41., 42., 43., 44., 45., 46., 47., 48., 49., Chris@87: 50., 51., 52., 53., 54., 55., 56., 57., 58., 59., Chris@87: 60., 61., 62., 63., 64., 65., 66., 67., 68., 69., Chris@87: 70., 71., 72., 73., 74., 75., 76., 77., 78., 79., Chris@87: 80., 81., 82., 83., 84., 85., 86., 87., 88., 89., Chris@87: 90., 91., 92., 93., 94., 95., 96., 97., 98., 99., Chris@87: Chris@87: 98., 98., 98., 98., 98., 98., 98., 98., 98., 98., Chris@87: 98., 98., 98., 98., 98., 98., 98., 98., 98., 98. Chris@87: ]) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_maximum_1(self): Chris@87: a = np.arange(100) Chris@87: a = pad(a, (25, 20), 'maximum') Chris@87: b = np.array( Chris@87: [99, 99, 99, 99, 99, 99, 99, 99, 99, 99, Chris@87: 99, 99, 99, 99, 99, 99, 99, 99, 99, 99, Chris@87: 99, 99, 99, 99, 99, Chris@87: Chris@87: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, Chris@87: 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, Chris@87: 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, Chris@87: 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, Chris@87: 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, Chris@87: 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, Chris@87: 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, Chris@87: 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, Chris@87: 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, Chris@87: 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, Chris@87: Chris@87: 99, 99, 99, 99, 99, 99, 99, 99, 99, 99, Chris@87: 99, 99, 99, 99, 99, 99, 99, 99, 99, 99] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_maximum_2(self): Chris@87: a = np.arange(100) + 1 Chris@87: a = pad(a, (25, 20), 'maximum') Chris@87: b = np.array( Chris@87: [100, 100, 100, 100, 100, 100, 100, 100, 100, 100, Chris@87: 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, Chris@87: 100, 100, 100, 100, 100, Chris@87: Chris@87: 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, Chris@87: 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, Chris@87: 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, Chris@87: 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, Chris@87: 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, Chris@87: 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, Chris@87: 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, Chris@87: 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, Chris@87: 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, Chris@87: 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, Chris@87: Chris@87: 100, 100, 100, 100, 100, 100, 100, 100, 100, 100, Chris@87: 100, 100, 100, 100, 100, 100, 100, 100, 100, 100] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_minimum_1(self): Chris@87: a = np.arange(100) Chris@87: a = pad(a, (25, 20), 'minimum') Chris@87: b = np.array( Chris@87: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, Chris@87: 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, Chris@87: 0, 0, 0, 0, 0, Chris@87: Chris@87: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, Chris@87: 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, Chris@87: 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, Chris@87: 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, Chris@87: 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, Chris@87: 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, Chris@87: 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, Chris@87: 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, Chris@87: 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, Chris@87: 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, Chris@87: Chris@87: 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, Chris@87: 0, 0, 0, 0, 0, 0, 0, 0, 0, 0] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_minimum_2(self): Chris@87: a = np.arange(100) + 2 Chris@87: a = pad(a, (25, 20), 'minimum') Chris@87: b = np.array( Chris@87: [2, 2, 2, 2, 2, 2, 2, 2, 2, 2, Chris@87: 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, Chris@87: 2, 2, 2, 2, 2, Chris@87: Chris@87: 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, Chris@87: 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, Chris@87: 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, Chris@87: 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, Chris@87: 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, Chris@87: 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, Chris@87: 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, Chris@87: 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, Chris@87: 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, Chris@87: 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, Chris@87: Chris@87: 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, Chris@87: 2, 2, 2, 2, 2, 2, 2, 2, 2, 2] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_median(self): Chris@87: a = np.arange(100).astype('f') Chris@87: a = pad(a, (25, 20), 'median') Chris@87: b = np.array( Chris@87: [49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, Chris@87: 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, Chris@87: 49.5, 49.5, 49.5, 49.5, 49.5, Chris@87: Chris@87: 0., 1., 2., 3., 4., 5., 6., 7., 8., 9., Chris@87: 10., 11., 12., 13., 14., 15., 16., 17., 18., 19., Chris@87: 20., 21., 22., 23., 24., 25., 26., 27., 28., 29., Chris@87: 30., 31., 32., 33., 34., 35., 36., 37., 38., 39., Chris@87: 40., 41., 42., 43., 44., 45., 46., 47., 48., 49., Chris@87: 50., 51., 52., 53., 54., 55., 56., 57., 58., 59., Chris@87: 60., 61., 62., 63., 64., 65., 66., 67., 68., 69., Chris@87: 70., 71., 72., 73., 74., 75., 76., 77., 78., 79., Chris@87: 80., 81., 82., 83., 84., 85., 86., 87., 88., 89., Chris@87: 90., 91., 92., 93., 94., 95., 96., 97., 98., 99., Chris@87: Chris@87: 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, Chris@87: 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_median_01(self): Chris@87: a = np.array([[3, 1, 4], [4, 5, 9], [9, 8, 2]]) Chris@87: a = pad(a, 1, 'median') Chris@87: b = np.array( Chris@87: [[4, 4, 5, 4, 4], Chris@87: Chris@87: [3, 3, 1, 4, 3], Chris@87: [5, 4, 5, 9, 5], Chris@87: [8, 9, 8, 2, 8], Chris@87: Chris@87: [4, 4, 5, 4, 4]] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_median_02(self): Chris@87: a = np.array([[3, 1, 4], [4, 5, 9], [9, 8, 2]]) Chris@87: a = pad(a.T, 1, 'median').T Chris@87: b = np.array( Chris@87: [[5, 4, 5, 4, 5], Chris@87: Chris@87: [3, 3, 1, 4, 3], Chris@87: [5, 4, 5, 9, 5], Chris@87: [8, 9, 8, 2, 8], Chris@87: Chris@87: [5, 4, 5, 4, 5]] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_mean_shape_one(self): Chris@87: a = [[4, 5, 6]] Chris@87: a = pad(a, (5, 7), 'mean', stat_length=2) Chris@87: b = np.array( Chris@87: [[4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6], Chris@87: [4, 4, 4, 4, 4, 4, 5, 6, 6, 6, 6, 6, 6, 6, 6]] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_mean_2(self): Chris@87: a = np.arange(100).astype('f') Chris@87: a = pad(a, (25, 20), 'mean') Chris@87: b = np.array( Chris@87: [49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, Chris@87: 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, Chris@87: 49.5, 49.5, 49.5, 49.5, 49.5, Chris@87: Chris@87: 0., 1., 2., 3., 4., 5., 6., 7., 8., 9., Chris@87: 10., 11., 12., 13., 14., 15., 16., 17., 18., 19., Chris@87: 20., 21., 22., 23., 24., 25., 26., 27., 28., 29., Chris@87: 30., 31., 32., 33., 34., 35., 36., 37., 38., 39., Chris@87: 40., 41., 42., 43., 44., 45., 46., 47., 48., 49., Chris@87: 50., 51., 52., 53., 54., 55., 56., 57., 58., 59., Chris@87: 60., 61., 62., 63., 64., 65., 66., 67., 68., 69., Chris@87: 70., 71., 72., 73., 74., 75., 76., 77., 78., 79., Chris@87: 80., 81., 82., 83., 84., 85., 86., 87., 88., 89., Chris@87: 90., 91., 92., 93., 94., 95., 96., 97., 98., 99., Chris@87: Chris@87: 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, Chris@87: 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5, 49.5] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: Chris@87: class TestConstant(TestCase): Chris@87: def test_check_constant(self): Chris@87: a = np.arange(100) Chris@87: a = pad(a, (25, 20), 'constant', constant_values=(10, 20)) Chris@87: b = np.array( Chris@87: [10, 10, 10, 10, 10, 10, 10, 10, 10, 10, Chris@87: 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, Chris@87: 10, 10, 10, 10, 10, Chris@87: Chris@87: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, Chris@87: 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, Chris@87: 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, Chris@87: 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, Chris@87: 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, Chris@87: 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, Chris@87: 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, Chris@87: 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, Chris@87: 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, Chris@87: 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, Chris@87: Chris@87: 20, 20, 20, 20, 20, 20, 20, 20, 20, 20, Chris@87: 20, 20, 20, 20, 20, 20, 20, 20, 20, 20] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: Chris@87: class TestLinearRamp(TestCase): Chris@87: def test_check_simple(self): Chris@87: a = np.arange(100).astype('f') Chris@87: a = pad(a, (25, 20), 'linear_ramp', end_values=(4, 5)) Chris@87: b = np.array( Chris@87: [4.00, 3.84, 3.68, 3.52, 3.36, 3.20, 3.04, 2.88, 2.72, 2.56, Chris@87: 2.40, 2.24, 2.08, 1.92, 1.76, 1.60, 1.44, 1.28, 1.12, 0.96, Chris@87: 0.80, 0.64, 0.48, 0.32, 0.16, Chris@87: Chris@87: 0.00, 1.00, 2.00, 3.00, 4.00, 5.00, 6.00, 7.00, 8.00, 9.00, Chris@87: 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, Chris@87: 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, Chris@87: 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, Chris@87: 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, Chris@87: 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, Chris@87: 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, Chris@87: 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, Chris@87: 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, Chris@87: 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, Chris@87: Chris@87: 94.3, 89.6, 84.9, 80.2, 75.5, 70.8, 66.1, 61.4, 56.7, 52.0, Chris@87: 47.3, 42.6, 37.9, 33.2, 28.5, 23.8, 19.1, 14.4, 9.7, 5.] Chris@87: ) Chris@87: assert_array_almost_equal(a, b, decimal=5) Chris@87: Chris@87: Chris@87: class TestReflect(TestCase): Chris@87: def test_check_simple(self): Chris@87: a = np.arange(100) Chris@87: a = pad(a, (25, 20), 'reflect') Chris@87: b = np.array( Chris@87: [25, 24, 23, 22, 21, 20, 19, 18, 17, 16, Chris@87: 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, Chris@87: 5, 4, 3, 2, 1, Chris@87: Chris@87: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, Chris@87: 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, Chris@87: 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, Chris@87: 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, Chris@87: 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, Chris@87: 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, Chris@87: 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, Chris@87: 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, Chris@87: 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, Chris@87: 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, Chris@87: Chris@87: 98, 97, 96, 95, 94, 93, 92, 91, 90, 89, Chris@87: 88, 87, 86, 85, 84, 83, 82, 81, 80, 79] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_large_pad(self): Chris@87: a = [[4, 5, 6], [6, 7, 8]] Chris@87: a = pad(a, (5, 7), 'reflect') Chris@87: b = np.array( Chris@87: [[7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7], Chris@87: Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7], Chris@87: Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7, 8, 7, 6, 7], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5]] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_shape(self): Chris@87: a = [[4, 5, 6]] Chris@87: a = pad(a, (5, 7), 'reflect') Chris@87: b = np.array( Chris@87: [[5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5], Chris@87: [5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5, 6, 5, 4, 5]] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_01(self): Chris@87: a = pad([1, 2, 3], 2, 'reflect') Chris@87: b = np.array([3, 2, 1, 2, 3, 2, 1]) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_02(self): Chris@87: a = pad([1, 2, 3], 3, 'reflect') Chris@87: b = np.array([2, 3, 2, 1, 2, 3, 2, 1, 2]) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_03(self): Chris@87: a = pad([1, 2, 3], 4, 'reflect') Chris@87: b = np.array([1, 2, 3, 2, 1, 2, 3, 2, 1, 2, 3]) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: Chris@87: class TestWrap(TestCase): Chris@87: def test_check_simple(self): Chris@87: a = np.arange(100) Chris@87: a = pad(a, (25, 20), 'wrap') Chris@87: b = np.array( Chris@87: [75, 76, 77, 78, 79, 80, 81, 82, 83, 84, Chris@87: 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, Chris@87: 95, 96, 97, 98, 99, Chris@87: Chris@87: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, Chris@87: 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, Chris@87: 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, Chris@87: 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, Chris@87: 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, Chris@87: 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, Chris@87: 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, Chris@87: 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, Chris@87: 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, Chris@87: 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, Chris@87: Chris@87: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, Chris@87: 10, 11, 12, 13, 14, 15, 16, 17, 18, 19] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_large_pad(self): Chris@87: a = np.arange(12) Chris@87: a = np.reshape(a, (3, 4)) Chris@87: a = pad(a, (10, 12), 'wrap') Chris@87: b = np.array( Chris@87: [[10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, Chris@87: 11, 8, 9, 10, 11, 8, 9, 10, 11], Chris@87: [2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, Chris@87: 3, 0, 1, 2, 3, 0, 1, 2, 3], Chris@87: [6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, Chris@87: 7, 4, 5, 6, 7, 4, 5, 6, 7], Chris@87: [10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, Chris@87: 11, 8, 9, 10, 11, 8, 9, 10, 11], Chris@87: [2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, Chris@87: 3, 0, 1, 2, 3, 0, 1, 2, 3], Chris@87: [6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, Chris@87: 7, 4, 5, 6, 7, 4, 5, 6, 7], Chris@87: [10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, Chris@87: 11, 8, 9, 10, 11, 8, 9, 10, 11], Chris@87: [2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, Chris@87: 3, 0, 1, 2, 3, 0, 1, 2, 3], Chris@87: [6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, Chris@87: 7, 4, 5, 6, 7, 4, 5, 6, 7], Chris@87: [10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, Chris@87: 11, 8, 9, 10, 11, 8, 9, 10, 11], Chris@87: Chris@87: [2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, Chris@87: 3, 0, 1, 2, 3, 0, 1, 2, 3], Chris@87: [6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, Chris@87: 7, 4, 5, 6, 7, 4, 5, 6, 7], Chris@87: [10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, Chris@87: 11, 8, 9, 10, 11, 8, 9, 10, 11], Chris@87: Chris@87: [2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, Chris@87: 3, 0, 1, 2, 3, 0, 1, 2, 3], Chris@87: [6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, Chris@87: 7, 4, 5, 6, 7, 4, 5, 6, 7], Chris@87: [10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, Chris@87: 11, 8, 9, 10, 11, 8, 9, 10, 11], Chris@87: [2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, Chris@87: 3, 0, 1, 2, 3, 0, 1, 2, 3], Chris@87: [6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, Chris@87: 7, 4, 5, 6, 7, 4, 5, 6, 7], Chris@87: [10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, Chris@87: 11, 8, 9, 10, 11, 8, 9, 10, 11], Chris@87: [2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, Chris@87: 3, 0, 1, 2, 3, 0, 1, 2, 3], Chris@87: [6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, Chris@87: 7, 4, 5, 6, 7, 4, 5, 6, 7], Chris@87: [10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, Chris@87: 11, 8, 9, 10, 11, 8, 9, 10, 11], Chris@87: [2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, Chris@87: 3, 0, 1, 2, 3, 0, 1, 2, 3], Chris@87: [6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, 7, 4, 5, 6, Chris@87: 7, 4, 5, 6, 7, 4, 5, 6, 7], Chris@87: [10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, 11, 8, 9, 10, Chris@87: 11, 8, 9, 10, 11, 8, 9, 10, 11]] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_01(self): Chris@87: a = pad([1, 2, 3], 3, 'wrap') Chris@87: b = np.array([1, 2, 3, 1, 2, 3, 1, 2, 3]) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: def test_check_02(self): Chris@87: a = pad([1, 2, 3], 4, 'wrap') Chris@87: b = np.array([3, 1, 2, 3, 1, 2, 3, 1, 2, 3, 1]) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: Chris@87: class TestStatLen(TestCase): Chris@87: def test_check_simple(self): Chris@87: a = np.arange(30) Chris@87: a = np.reshape(a, (6, 5)) Chris@87: a = pad(a, ((2, 3), (3, 2)), mode='mean', stat_length=(3,)) Chris@87: b = np.array( Chris@87: [[6, 6, 6, 5, 6, 7, 8, 9, 8, 8], Chris@87: [6, 6, 6, 5, 6, 7, 8, 9, 8, 8], Chris@87: Chris@87: [1, 1, 1, 0, 1, 2, 3, 4, 3, 3], Chris@87: [6, 6, 6, 5, 6, 7, 8, 9, 8, 8], Chris@87: [11, 11, 11, 10, 11, 12, 13, 14, 13, 13], Chris@87: [16, 16, 16, 15, 16, 17, 18, 19, 18, 18], Chris@87: [21, 21, 21, 20, 21, 22, 23, 24, 23, 23], Chris@87: [26, 26, 26, 25, 26, 27, 28, 29, 28, 28], Chris@87: Chris@87: [21, 21, 21, 20, 21, 22, 23, 24, 23, 23], Chris@87: [21, 21, 21, 20, 21, 22, 23, 24, 23, 23], Chris@87: [21, 21, 21, 20, 21, 22, 23, 24, 23, 23]] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: Chris@87: class TestEdge(TestCase): Chris@87: def test_check_simple(self): Chris@87: a = np.arange(12) Chris@87: a = np.reshape(a, (4, 3)) Chris@87: a = pad(a, ((2, 3), (3, 2)), 'edge') Chris@87: b = np.array( Chris@87: [[0, 0, 0, 0, 1, 2, 2, 2], Chris@87: [0, 0, 0, 0, 1, 2, 2, 2], Chris@87: Chris@87: [0, 0, 0, 0, 1, 2, 2, 2], Chris@87: [3, 3, 3, 3, 4, 5, 5, 5], Chris@87: [6, 6, 6, 6, 7, 8, 8, 8], Chris@87: [9, 9, 9, 9, 10, 11, 11, 11], Chris@87: Chris@87: [9, 9, 9, 9, 10, 11, 11, 11], Chris@87: [9, 9, 9, 9, 10, 11, 11, 11], Chris@87: [9, 9, 9, 9, 10, 11, 11, 11]] Chris@87: ) Chris@87: assert_array_equal(a, b) Chris@87: Chris@87: Chris@87: class TestZeroPadWidth(TestCase): Chris@87: def test_zero_pad_width(self): Chris@87: arr = np.arange(30) Chris@87: arr = np.reshape(arr, (6, 5)) Chris@87: for pad_width in (0, (0, 0), ((0, 0), (0, 0))): Chris@87: assert_array_equal(arr, pad(arr, pad_width, mode='constant')) Chris@87: Chris@87: Chris@87: class ValueError1(TestCase): Chris@87: def test_check_simple(self): Chris@87: arr = np.arange(30) Chris@87: arr = np.reshape(arr, (6, 5)) Chris@87: kwargs = dict(mode='mean', stat_length=(3, )) Chris@87: assert_raises(ValueError, pad, arr, ((2, 3), (3, 2), (4, 5)), Chris@87: **kwargs) Chris@87: Chris@87: def test_check_negative_stat_length(self): Chris@87: arr = np.arange(30) Chris@87: arr = np.reshape(arr, (6, 5)) Chris@87: kwargs = dict(mode='mean', stat_length=(-3, )) Chris@87: assert_raises(ValueError, pad, arr, ((2, 3), (3, 2)), Chris@87: **kwargs) Chris@87: Chris@87: def test_check_negative_pad_width(self): Chris@87: arr = np.arange(30) Chris@87: arr = np.reshape(arr, (6, 5)) Chris@87: kwargs = dict(mode='mean', stat_length=(3, )) Chris@87: assert_raises(ValueError, pad, arr, ((-2, 3), (3, 2)), Chris@87: **kwargs) Chris@87: Chris@87: Chris@87: class ValueError2(TestCase): Chris@87: def test_check_simple(self): Chris@87: arr = np.arange(30) Chris@87: arr = np.reshape(arr, (6, 5)) Chris@87: kwargs = dict(mode='mean', stat_length=(3, )) Chris@87: assert_raises(ValueError, pad, arr, ((2, 3, 4), (3, 2)), Chris@87: **kwargs) Chris@87: Chris@87: Chris@87: class ValueError3(TestCase): Chris@87: def test_check_simple(self): Chris@87: arr = np.arange(30) Chris@87: arr = np.reshape(arr, (6, 5)) Chris@87: kwargs = dict(mode='mean', stat_length=(3, )) Chris@87: assert_raises(ValueError, pad, arr, ((-2, 3), (3, 2)), Chris@87: **kwargs) Chris@87: Chris@87: Chris@87: if __name__ == "__main__": Chris@87: run_module_suite()