Пример #1
0
def test_make_stock_from_plain_file(shared_datadir, tmpdir, add_cli_arguments, stock):
    output_name = str(tmpdir / "temp.hdf5")
    filename = str(shared_datadir / "smiles_plain.txt")
    add_cli_arguments(f"--files {filename} --output {output_name}")

    make_stock_main()

    stock.load_stock(filename, "stock1")
    stock.select_stocks(["stock1"])
    assert len(stock) == 3
Пример #2
0
def test_make_stock_from_plain_file(create_dummy_smiles_source, tmpdir,
                                    add_cli_arguments, default_config):
    output_name = str(tmpdir / "temp.hdf5")
    filename = create_dummy_smiles_source("txt")
    add_cli_arguments(f"--files {filename} --output {output_name}")

    make_stock_main()

    default_config.stock.load(filename, "stock1")
    default_config.stock.select(["stock1"])
    assert len(default_config.stock) == 3