def test_cat_main(shared_datadir, tmpdir, add_cli_arguments):
    filename = str(tmpdir / "output.hdf")
    inputs = [str(shared_datadir / "stock1.h5"), str(shared_datadir / "stock2.h5")]
    add_cli_arguments(f"--files {inputs[0]} {inputs[1]} --output {filename}")

    cat_main()

    data = pd.read_hdf(filename, "table")
    assert len(data) == 4
Exemple #2
0
def test_cat_main(tmpdir, add_cli_arguments, create_dummy_stock1,
                  create_dummy_stock2):
    filename = str(tmpdir / "output.hdf")
    inputs = [create_dummy_stock1("hdf5"), create_dummy_stock2]
    add_cli_arguments(f"--files {inputs[0]} {inputs[1]} --output {filename}")

    cat_main()

    data = pd.read_hdf(filename, "table")
    assert len(data) == 4