def test_into_DataFrame_concat(): csv = CSV(os.path.join(os.path.dirname(__file__), 'accounts.csv')) df = into(pd.DataFrame, Concat([csv, csv])) csv_df = csv.pandas_read_csv() assert df.index.tolist() == list(range(len(df))) assert df.values.tolist() == (csv_df.values.tolist() + csv_df.values.tolist()) assert df.columns.tolist() == csv_df.columns.tolist()
def test_datetime_csv_reader_same_as_into_types(): csv = CSV(os.path.join(os.path.dirname(__file__), 'accounts.csv')) rhs = csv.pandas_read_csv().dtypes df = into(pd.DataFrame, csv) dtypes = df.dtypes expected = pd.Series( [np.dtype(x) for x in ['i8', 'i8', 'O', 'datetime64[ns]']], index=csv.columns) assert dtypes.index.tolist() == expected.index.tolist() assert dtypes.tolist() == expected.tolist()
def test_datetime_csv_reader_same_as_into_types(): csv = CSV(os.path.join(os.path.dirname(__file__), 'accounts.csv')) rhs = csv.pandas_read_csv().dtypes df = into(pd.DataFrame, csv) dtypes = df.dtypes expected = pd.Series([np.dtype(x) for x in ['i8', 'i8', 'O', 'datetime64[ns]']], index=csv.columns) assert dtypes.index.tolist() == expected.index.tolist() assert dtypes.tolist() == expected.tolist()
def test_datetime_csv_reader_same_as_into(): csv = CSV(os.path.join(os.path.dirname(__file__), 'accounts.csv')) rhs = csv.pandas_read_csv().dtypes df = into(pd.DataFrame, csv) dtypes = df.dtypes expected = pd.Series( [np.dtype(x) for x in ['i8', 'i8', 'O', 'datetime64[ns]']], index=csv.columns) # make sure reader with no args does the same thing as into() # Values the same assert dtypes.index.tolist() == rhs.index.tolist() assert dtypes.tolist() == rhs.tolist()
def test_datetime_csv_reader_same_as_into(): csv = CSV(os.path.join(os.path.dirname(__file__), 'accounts.csv')) rhs = csv.pandas_read_csv().dtypes df = into(pd.DataFrame, csv) dtypes = df.dtypes expected = pd.Series([np.dtype(x) for x in ['i8', 'i8', 'O', 'datetime64[ns]']], index=csv.columns) # make sure reader with no args does the same thing as into() # Values the same assert dtypes.index.tolist() == rhs.index.tolist() assert dtypes.tolist() == rhs.tolist()