def test_set_item_function(self):
df = pandas.DataFrame(data=dict(a=[4.5], b=[6], c=[7]))
self.assertRaise(lambda: StreamingDataFrame(df), TypeError)
sdf = StreamingDataFrame.read_df(df)
sdf['bb'] = sdf['b'].apply(lambda x: x + 11)
df = sdf.to_df()
ddf = ddf = pandas.DataFrame(
data=dict(a=[4.5], b=[6], c=[7], bb=[17]))
self.assertEqualDataFrame(df, ddf)
def test_set_item(self):
df = pandas.DataFrame(data=dict(a=[4.5], b=[6], c=[7]))
self.assertRaise(lambda: StreamingDataFrame(df), TypeError)
sdf = StreamingDataFrame.read_df(df)
def f():
sdf[['a']] = 10
self.assertRaise(f, ValueError)
def g():
sdf['a'] = [10]
self.assertRaise(g, NotImplementedError)
sdf['aa'] = 10
df = sdf.to_df()
ddf = pandas.DataFrame(data=dict(a=[4.5], b=[6], c=[7], aa=[10]))
self.assertEqualDataFrame(df, ddf)
sdf['bb'] = sdf['b'] + 10
df = sdf.to_df()
ddf = ddf = pandas.DataFrame(
data=dict(a=[4.5], b=[6], c=[7], aa=[10], bb=[16]))
self.assertEqualDataFrame(df, ddf)
def test_init(self):
sdf = dummy_streaming_dataframe(100)
df1 = sdf.to_df()
sdf2 = StreamingDataFrame(sdf)
df2 = sdf2.to_df()
self.assertEqualDataFrame(df1, df2)