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_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_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_with_objects_raises():
class Example(object):
def __init__(self, data):
self.data = data
def get(self):
return self.data
f = FrameConverter()
df = pd.DataFrame(data={'one': [Example(444)]})
with pytest.raises(Exception):
f.docify(df)
def test_with_objects_raises():
class Example(object):
def __init__(self, data):
self.data = data
def get(self):
return self.data
f = FrameConverter()
df = pd.DataFrame(data={'one': [Example(444)]})
with pytest.raises(Exception):
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_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)
