def test_row_labels_label(self): df = pd.DataFrame(data=np.arange(10).reshape((5, 2))) s = pd.Series(['a', 'b', 'c', 'd', 'e']) df.insert(loc=0, column='hey', value=s) assert_frame_equal( make_array_into_dataframe(data=np.arange(10).reshape((5, 2)), row_labels=['a', 'b', 'c', 'd', 'e'], row_labels_label='hey'), df)
def test_row_labels(self): df = pd.DataFrame(data=np.arange(10).reshape((5,2))) s = pd.Series(['a', 'b', 'c', 'd', 'e']) df.insert(loc=0, column='item', value=s) assert_frame_equal(make_array_into_dataframe( data=np.arange(10).reshape((5,2)), row_labels=['a', 'b', 'c', 'd', 'e']), df)
def test_column_labels(self): assert_frame_equal( make_array_into_dataframe(data=np.arange(10).reshape((5, 2)), column_labels=['a', 'b']), pd.DataFrame(data=np.arange(10).reshape((5, 2)), columns=['a', 'b']))
def test_defaults(self): assert_frame_equal( make_array_into_dataframe(data=np.arange(10).reshape((5, 2))), pd.DataFrame(data=np.arange(10).reshape((5, 2))))
def test_column_labels(self): assert_frame_equal(make_array_into_dataframe( data=np.arange(10).reshape((5,2)), column_labels=['a', 'b']), pd.DataFrame(data=np.arange(10).reshape((5,2)), columns=['a', 'b']))
def test_defaults(self): assert_frame_equal(make_array_into_dataframe( data=np.arange(10).reshape((5,2))), pd.DataFrame(data=np.arange(10).reshape((5,2))))