Skip to content

Commit 1631836

Browse files
authored
test dataframe dir
1 parent 0716a8f commit 1631836

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

streamz/dataframe/tests/test_dataframes.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1007,3 +1007,10 @@ def test_windowed_groupby_aggs_with_start_state(stream):
10071007
stream.emit(df)
10081008
out_df1 = pd.DataFrame({'name':['Alice', 'Bob', 'Linda', 'Tom'], 'amount':[50.0, 550.0, 100.0, 150.0]})
10091009
assert_eq(output1[-1][1].reset_index(), out_df1)
1010+
1011+
1012+
def test_dir(stream):
1013+
example = pd.DataFrame({'name': [], 'amount': []})
1014+
sdf = DataFrame(stream, example=example)
1015+
assert 'name' in dir(sdf)
1016+
assert 'amount' in dir(sdf)

0 commit comments

Comments
 (0)