def test_np_5d_set_axes_sorted(self): x = np.zeros([6, 2, 5, 3, 4]) shape = None axes = [4, 1, 2] shape_expected = np.array([2, 5, 4]) axes_expected = np.array([1, 2, 4]) shape_res, axes_res = _init_nd_shape_and_axes_sorted(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected)
def test_np_5d_set_shape_axes_sorted(self): x = np.zeros([6, 2, 5, 3, 4]) shape = [10, -1, 2] axes = [1, 0, 3] shape_expected = np.array([6, 10, 2]) axes_expected = np.array([0, 1, 3]) shape_res, axes_res = _init_nd_shape_and_axes_sorted(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected)
def test_np_5d_defaults(self): x = np.zeros([6, 2, 5, 3, 4]) shape = None axes = None shape_expected = np.array([6, 2, 5, 3, 4]) axes_expected = np.array([0, 1, 2, 3, 4]) shape_res, axes_res = _init_nd_shape_and_axes(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected) shape_res, axes_res = _init_nd_shape_and_axes_sorted(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected)
def test_np_2d_defaults(self): x = np.arange(0, 1, .1).reshape(5, 2) shape = None axes = None shape_expected = np.array([5, 2]) axes_expected = np.array([0, 1]) shape_res, axes_res = _init_nd_shape_and_axes(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected) shape_res, axes_res = _init_nd_shape_and_axes_sorted(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected)
def test_py_2d_defaults(self): x = [[1, 2, 3, 4], [5, 6, 7, 8]] shape = None axes = None shape_expected = np.array([2, 4]) axes_expected = np.array([0, 1]) shape_res, axes_res = _init_nd_shape_and_axes(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected) shape_res, axes_res = _init_nd_shape_and_axes_sorted(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected)
def test_py_1d_defaults(self): x = [1, 2, 3] shape = None axes = None shape_expected = np.array([3]) axes_expected = np.array([0]) shape_res, axes_res = _init_nd_shape_and_axes(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected) shape_res, axes_res = _init_nd_shape_and_axes_sorted(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected)
def test_np_0d_defaults(self): x = np.array(7.) shape = None axes = None shape_expected = np.array([]) axes_expected = np.array([]) shape_res, axes_res = _init_nd_shape_and_axes(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected) shape_res, axes_res = _init_nd_shape_and_axes_sorted(x, shape, axes) assert_equal(shape_res, shape_expected) assert_equal(axes_res, axes_expected)