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)