from File import *
from Plot import *


def test_bar():
    men_means = [8457.031, 26788.493, 44742.531, 61611.609, 81512.44908]
    same_color_bar([men_means, men_means, men_means], [0, 20000, 40000, 60000, 80000, 100000],
                   "resources/img/bpfl/test2.png")


def test_plot():
    base_path = "./resources/data/"
    file_name = ["BFLC无恶意.log", "BFLC10恶意.log", "BFLC20恶意.log", "BFLC30恶意.log", "BFLC40恶意.log", "BFLC50恶意.log"]
    plot_value = []
    for name in file_name:
        plot_value.append(read_acc_list(base_path + name))
    multi_plot(plot_value, [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7], "resources/img/bpfl/BFLC.png")


if __name__ == '__main__':
    test_plot()