def test_length_and_shape_when_passing_multiple_lists(): ys = [[1, 222, 3, -3.14, 0], [1, 222, 3, -3.14, 0, 4, 4, 4], [1]] xs = [[1000, 222, 3, -314, 0], [0, 2222, 3, -3.14, 0, 4, 4, 4], [1]] series = MultiSeries(xs=xs, ys=ys) assert len(series) == 3 assert series.shape() == [5, 8, 1]
def test_length_and_shape_when_only_passing_single_ys_as_list(): ys = [1, 2, 3] series = MultiSeries(ys=ys) assert len(series) == 1 assert series.shape() == [3]