Skip to content

Commit 0286d01

Browse files
committed
Fix up tests
1 parent 0b1b8ed commit 0286d01

1 file changed

Lines changed: 5 additions & 10 deletions

File tree

streamz/dataframe/tests/test_dataframes.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -713,26 +713,21 @@ def test_windowing_n(func, n, getter):
713713
lambda x: x.sum(),
714714
lambda x: x.mean(),
715715
lambda x: x.count(),
716-
lambda x: x.size,
717716
lambda x: x.var(ddof=1),
718717
lambda x: x.std(ddof=1),
719718
lambda x: x.var(ddof=0),
720-
])
721-
@pytest.mark.parametrize('getter', [
722-
lambda df: df,
723-
lambda df: df.x,
724-
])
725-
def test_expanding(func, getter):
719+
], ids=["sum", "mean", "count", "var_1", "std", "var_0"])
720+
def test_expanding(func):
726721
df = pd.DataFrame({'x': [1.], 'y': [2.]})
727722
sdf = DataFrame(example=df)
728723

729-
L = func(getter(sdf).expanding()).stream.gather().sink_to_list()
724+
L = func(sdf.expanding()).stream.gather().sink_to_list()
730725

731726
for i in range(5):
732727
sdf.emit(df)
733728

734-
result = pd.concat(L, axis=1).T
735-
expected = func(getter(pd.concat([df] * 5)).expanding())
729+
result = pd.concat(L, axis=1).T.astype(float)
730+
expected = func(pd.concat([df] * 5, ignore_index=True).expanding())
736731
assert_eq(result, expected)
737732

738733

0 commit comments

Comments
 (0)