def test_to_pandas_dataframe_using_columns(self): sds = SimpleDataSource(schema=[('a', int32), ('b', str)], columns={'a': [1, 2, 3], 'b': ['one', 'two', 'three']}) df = sds.to_pandas_dataframe() result = repr(df) self.assertEquals(expected_repr_123, result)
def test_to_pandas_dataframe_using_rows(self): sds = SimpleDataSource(schema=[('a', int32), ('b', str)], rows=[(1, 'one'), (2, 'two'), (3, 'three')]) df = sds.to_pandas_dataframe() result = repr(df) self.assertEquals(expected_repr_123, result)