def test_correct_invertcolumns_type(self, dataframe_wide):
with pytest.raises(TypeError):
gather(
df=dataframe_wide,
key="year",
value="actual",
columns=["2018", "2019"],
invert_columns="Yes",
)
def test_equal_df(self, dataframe_wide):
df1 = gather(
df=dataframe_wide,
key="year",
value="actual",
columns=["country", "continent"],
invert_columns=True,
# convert=True,
)
df2 = gather(
df=dataframe_wide, key="year", value="actual", columns=["2018", "2019"]
)
assert df1.equals(df2)
def test_gather(self, dataframe_wide):
__df = gather(
df=dataframe_wide,
key="year",
value="actual",
columns=range(2, 4),
invert_columns=False,
drop_na=True,
# convert=True,
)
def test_correct_column_type(self, dataframe_wide):
with pytest.raises(TypeError):
gather(df=dataframe_wide, key="year", value="actual", columns="string")
def test_correct_length_range(self, dataframe_wide):
with pytest.raises(IndexError):
gather(df=dataframe_wide, key="year", value="actual", columns=range(2, 100))