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))
