def test_create_column_2d(): shape = 2, 3 result = sqlite._create_colnames(shape) expected = ['v1_1', 'v1_2', 'v1_3', 'v2_1', 'v2_2', 'v2_3'] assert result == expected assert sqlite._create_shape(result) == shape
def test_create_column_3d(): shape = 2, 3, 4 assert sqlite._create_shape(sqlite._create_colnames(shape)) == shape
def test_create_column_1d(): shape = 2, result = sqlite._create_colnames(shape) expected = ['v1', 'v2'] assert result == expected assert sqlite._create_shape(result) == shape