def test_generate_augmented_data_sparse(self):
x = csr_matrix(np.zeros((3, 6)))
x_augmented = _generate_augmented_data(x)
assert x_augmented.shape[0] == 6 and x_augmented.shape[1] == 6
def test_generate_augmented_data_ndarray(self):
x = np.ones((3, 6))
x_augmented = _generate_augmented_data(x)
assert x_augmented.shape[0] == 6 and x_augmented.shape[1] == 6