def test_compare_all_empty():
"""Регрессионный тест на ошибку в сравнении пустых DataFrame."""
df = div_status._compare(
pd.DataFrame(columns=[1]),
pd.DataFrame(columns=[2]),
)
pd.testing.assert_frame_equal(
df,
pd.DataFrame(columns=[1, 2, "STATUS"]),
)
def test_compare():
"""Сравнение и распечатка результатов."""
df = div_status._compare(
pd.DataFrame([1, 2], columns=["a"]),
pd.DataFrame([1, 3], columns=["b"]),
)
pd.testing.assert_frame_equal(
df,
pd.DataFrame(
[[1, 1, ""], [2, 3, "ERROR"]],
columns=["a", "b", "STATUS"],
),
)
def test_compare_with_empty():
"""Регрессионный тест на ошибку в сравнении с пустым DataFrame."""
df = div_status._compare(
pd.DataFrame([1, 2], columns=[3]),
pd.DataFrame(columns=[1]),
)
pd.testing.assert_frame_equal(
df,
pd.DataFrame(
[[1, None, "ERROR"], [2, None, "ERROR"]],
columns=[3, 1, "STATUS"],
index=[0, 1],
),
check_index_type=False,
)
def test_compare(capsys):
"""Сравнение и распечатка результатов."""
df = div_status._compare(
"test_name",
pd.DataFrame([1, 2]),
pd.DataFrame([1, 3]),
)
pd.testing.assert_frame_equal(
df,
pd.DataFrame(
[[1, 1, ""], [2, 3, "ERROR"]],
columns=["LOCAL", "SOURCE", "STATUS"],
),
)
captured = capsys.readouterr()
assert "test_name" in captured.out