def test_to_records_noindex(): df = DataFrame() df['a'] = aa = np.arange(10, dtype=np.int32) df['b'] = bb = np.arange(10, 20, dtype=np.float64) rec = df.to_records(index=False) assert rec.dtype.names == ('a', 'b') np.testing.assert_array_equal(rec['a'], aa) np.testing.assert_array_equal(rec['b'], bb)
def test_to_records_withindex(): df = DataFrame() df['a'] = aa = np.arange(10, dtype=np.int32) df['b'] = bb = np.arange(10, 20, dtype=np.float64) rec_indexed = df.to_records(index=True) assert rec_indexed.size == len(aa) assert rec_indexed.dtype.names == ('index', 'a', 'b') np.testing.assert_array_equal(rec_indexed['a'], aa) np.testing.assert_array_equal(rec_indexed['b'], bb) np.testing.assert_array_equal(rec_indexed['index'], np.arange(10))