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