def test_equal_column_dtype_index_iris():
iris, columns, dtype, index = _iris()
iris_dummies = _iris_dummies(prefix_sep="#")
new_iris = from_dummies(
iris_dummies, prefix_sep="#", column_order=columns, dtypes=dtype, index=index
)
assert new_iris.equals(iris) == True
def test_not_equal_column_dtype_iris():
iris, columns, dtype, _ = _iris()
iris_dummies = _iris_dummies(prefix_sep="#")
new_iris = from_dummies(
iris_dummies, prefix_sep="#", column_order=columns, dtypes=dtype
)
assert new_iris.equals(iris) == False
def test_equal_basic():
dummies = pd.get_dummies(df_basic, prefix=prefix_basic)
new_df = from_dummies(df_basic, column_order=columns_basic)
assert new_df.equals(new_df) == True
def test_not_equal_iris():
iris, _, __, ___ = _iris()
print(iris.head())
iris_dummies = _iris_dummies(prefix_sep="#")
new_iris = from_dummies(iris_dummies, prefix_sep="#")
assert new_iris.equals(iris) == False
def test_type_basic():
dummies = pd.get_dummies(df_basic, prefix=prefix_basic)
assert type(from_dummies(dummies_basic)) == pd.core.frame.DataFrame