def test_0d_1c(self):
input = 4
output = adaptDim(input, 1)
expected = [4]
nptest.assert_array_equal(output, expected)
def test_2d_2c(self):
input = [[1, 1], [2, 2], [3, 3], [4, 4]]
output = adaptDim(input, 2)
expected = input
nptest.assert_array_equal(output, expected)
def test_2d_1c(self):
input = [[1, 1], [2, 2], [3, 3], [4, 4]]
output = adaptDim(input, 1)
expected = np.array([1, 2, 3, 4]).reshape(-1, 1)
nptest.assert_array_equal(output, expected)
def test_1d_1c(self):
input = np.array([1, 2, 3, 4]).reshape(-1, 1)
output = adaptDim(input, 1)
expected = input
nptest.assert_array_equal(output, expected)