def test_unite_round_trip_grouped_df():
data = data_frame(label=["s1_e1", "s2_e2"])
separated = separate(data.groupby('label'),
"label",
into=["season", "episode"])
united = unite(separated, "label", "season", "episode")
assert_frame_sort_equal(united, data)
def test_separate_remove_arg(df):
assert_equal_query(
df, separate("label", into=["season", "episode"], remove=False),
df.assign(season=["S1", "S1"], episode=["E1", "E2"]))
def test_separate_warn_arg_warn():
data = data_frame(label="1-2-3-4")
with pytest.warns(UserWarning):
separate(data, "label", into=["a", "b"], sep="-")
def test_separate_convert_arg():
data = data_frame(label=["1-1", "2-a"])
assert_equal_query(
data, separate("label", into=["season", "episode"], convert=True),
data_frame(season=[1, 2], episode=["1", "a"]))
def test_separate_sep_arg(df):
assert_equal_query(
df,
separate("label", into=["season", "episode"], sep="E"),
data_frame(season=["S1-", "S1-"], episode=["1", "2"]),
)
def test_separate_default(df):
assert_equal_query(df, separate("label", into=["season", "episode"]),
data_frame(season=["S1", "S1"], episode=["E1", "E2"]))