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