def test_numpy_non_contiguious(): recdtype = np.dtype([("index", np.int64), ("a", np.int32)]) rec = np.recarray(10, dtype=recdtype) rec.index = np.arange(30, 40) rec.a = aa = np.arange(20, dtype=np.int32)[::2] assert rec.a.flags["C_CONTIGUOUS"] is False gdf = DataFrame.from_records(rec, index="index") np.testing.assert_array_equal(aa, gdf["a"].to_array())
def test_numpy_non_contiguious(): recdtype = np.dtype([ ('index', np.int64), ('a', np.int32), ]) rec = np.recarray(10, dtype=recdtype) rec.index = np.arange(30, 40) rec.a = aa = np.arange(20, dtype=np.int32)[::2] assert rec.a.flags['C_CONTIGUOUS'] is False gdf = DataFrame.from_records(rec, index='index') np.testing.assert_array_equal(aa, gdf['a'].to_array())
def test_from_records_noindex(columns): recdtype = np.dtype([("a", np.int32), ("b", np.float64)]) rec = np.recarray(10, dtype=recdtype) rec.a = aa = np.arange(10, dtype=np.int32) rec.b = bb = np.arange(10, 20, dtype=np.float64) df = DataFrame.from_records(rec, columns=columns) if columns and "a" in columns: np.testing.assert_array_equal(aa, df["a"]) if columns and "b" in columns: np.testing.assert_array_equal(bb, df["b"]) np.testing.assert_array_equal(np.arange(10), df.index.values)
def test_from_records_withindex(columns): recdtype = np.dtype([ ('index', np.int64), ('a', np.int32), ('b', np.float64), ]) rec = np.recarray(10, dtype=recdtype) rec.index = ii = np.arange(30, 40) rec.a = aa = np.arange(10, dtype=np.int32) rec.b = bb = np.arange(10, 20, dtype=np.float64) df = DataFrame.from_records(rec, index='index') if columns and 'a' in columns: np.testing.assert_array_equal(aa, df['a']) if columns and 'b' in columns: np.testing.assert_array_equal(bb, df['b']) np.testing.assert_array_equal(ii, df.index.values)