import unittest
from pathlib import Path
from concho.config import HtmlConfigurator
from concho.hpev2 import HpeV2ConfiguratorParser
from concho.procs_chooser import plot_configurators
from concho.procs_chooser import ConfigPrice
from concho.procs_chooser import ConfigFlopsPerEuro


class Test(unittest.TestCase):

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.plots_dir = Path('./plots/')
        self.plots_dir.mkdir(exist_ok=True, parents=True)

    def test_hpe_bpu11_configs(self):
        configurators = [
            HtmlConfigurator(Path('catalogs/hpev2/20250314-cat2-conf16-hpe-dl380-gen11.html'), HpeV2ConfiguratorParser()),
            HtmlConfigurator(Path('catalogs/hpev2/20250314-cat2-conf19-hpe-dl365-gen11.html'), HpeV2ConfiguratorParser()),
        ]

        def config_filter(config):
            return True  # config.get_price() < 40000.0

        plot_configurators(configurators=configurators, ram_per_core=4.0e9, xaxis_def=ConfigPrice(), yaxis_def=ConfigFlopsPerEuro(), plot_title='physmol/dbossion ais configs', config_filter=config_filter, figure_file_path=self.plots_dir / '2025-dbossion-ais-configs.pdf')  # './hpe-2025-03-14.pdf')


if __name__ == '__main__':
    unittest.main()