def test_reduce_to_2d(self):
def make_range_array(dim):
"""Produce an incrementally increasing multidimensional array.
Args:
dim: the number of dimensions for the array
Returns:
An array of increasing integer elements, with dim dimensions and size
two in each dimension.
Example: rangeArray(2) results in [[0,1],[2,3]].
"""
return np.array(range(2**dim)).reshape([2] * dim)
for i in range(2, 5):
actual = text_plugin.reduce_to_2d(make_range_array(i))
expected = make_range_array(2)
np.testing.assert_array_equal(actual, expected)
def test_reduce_to_2d(self):
def make_range_array(dim):
"""Produce an incrementally increasing multidimensional array.
Args:
dim: the number of dimensions for the array
Returns:
An array of increasing integer elements, with dim dimensions and size
two in each dimension.
Example: rangeArray(2) results in [[0,1],[2,3]].
"""
return np.array(range(2**dim)).reshape([2] * dim)
for i in range(2, 5):
actual = text_plugin.reduce_to_2d(make_range_array(i))
expected = make_range_array(2)
np.testing.assert_array_equal(actual, expected)
