def test_frame_converter_with_some_invalid_column_subset(): f = FrameConverter() df = pd.DataFrame(np.random.randint(0, 100, size=(100, 4)), columns=list('ABCD')) expected = pd.DataFrame({'A': df['A'], 'N': np.nan}) assert_frame_equal(f.objify(f.docify(df), columns=['A', 'N']), expected[['A', 'N']])
def test_dataframe_writable_after_objify(): f = FrameConverter() df = pd.DataFrame(data={'one': [5, 6, 2]}) df = f.objify(f.docify(df)) df['one'] = 7 assert np.all(df['one'].values == np.array([7, 7, 7]))
def test_frame_converter_with_all_valid_column_subset(): f = FrameConverter() df = pd.DataFrame(np.random.randint(0, 100, size=(100, 4)), columns=list('ABCD')) assert_frame_equal(f.objify(f.docify(df), columns=['A', 'B']), df[['A', 'B']])
def test_dataframe_writable_after_objify(): f = FrameConverter() df = pd.DataFrame(data={'one': [5, 6, 2]}) df = f.objify(f.docify(df)) df['one'] = 7 assert np.all(df['one'].values == np.array([7, 7, 7]))
def test_frame_converter(): f = FrameConverter() df = pd.DataFrame(np.random.randint(0, 100, size=(100, 4)), columns=list('ABCD')) assert_frame_equal(f.objify(f.docify(df)), df)
def test_frame_converter_with_no_valid_column_subset(): f = FrameConverter() df = pd.DataFrame(np.random.randint(0, 100, size=(100, 4)), columns=list('ABCD')) assert f.objify(f.docify(df), columns=['N']).empty
def test_with_strings(): f = FrameConverter() df = pd.DataFrame(data={'one': ['a', 'b', 'c']}) assert_frame_equal(f.objify(f.docify(df)), df)
def test_objify_with_missing_columns(): f = FrameConverter() df = pd.DataFrame(data={'one': ['a', 'b', 'c', np.NaN]}) res = f.objify(f.docify(df), columns=['one', 'two']) assert res['one'].equals(df['one']) assert all(res['two'].isnull())
def test_string_cols_with_nans(): f = FrameConverter() df = pd.DataFrame(data={'one': ['a', 'b', 'c', np.NaN]}) assert (df.equals(f.objify(f.docify(df))))
def test_frame_converter(): f = FrameConverter() df = pd.DataFrame(np.random.randint(0, 100, size=(100, 4)), columns=list('ABCD')) assert_frame_equal(f.objify(f.docify(df)), df)
def test_string_cols_with_nans(): f = FrameConverter() df = pd.DataFrame(data={'one': ['a', 'b', 'c', np.NaN]}) assert(df.equals(f.objify(f.docify(df))))
def test_with_strings(): f = FrameConverter() df = pd.DataFrame(data={'one': ['a', 'b', 'c']}) assert_frame_equal(f.objify(f.docify(df)), df)