def test_main(datadir):
"""Test command line interface"""
# Test installation
assert os.system("csv2ofmvol -h") == 0
sys.argv = ["csv2ofmvol", "prodA3.csv", "prodA4.csv", "-o", "outfile.vol"]
csv2ofmvol.main()
vollines = open("outfile.vol").readlines()
assert sum(["*NAME" in line for line in vollines]) == 2
assert sum(["*METRIC" in line for line in vollines]) == 1
assert sum(["NAME A-3" in line for line in vollines]) == 1
assert sum(["NAME A-4" in line for line in vollines]) == 1
assert sum(["*OIL" in line for line in vollines]) == 1
def test_main(datadir, mocker):
"""Test command line interface"""
# Test installation
assert subprocess.check_output(["csv2ofmvol", "-h"])
mocker.patch(
"sys.argv",
["csv2ofmvol", "prodA3.csv", "prodA4.csv", "-o", "outfile.vol"])
csv2ofmvol.main()
vollines = open("outfile.vol").readlines()
assert sum(["*NAME" in line for line in vollines]) == 2
assert sum(["*METRIC" in line for line in vollines]) == 1
assert sum(["NAME A-3" in line for line in vollines]) == 1
assert sum(["NAME A-4" in line for line in vollines]) == 1
assert sum(["*OIL" in line for line in vollines]) == 1
def test_roundtrip(datadir):
"""Test that ofmvol2csv and csv2ofmvol can work together as inverses
of each other."""
ofmvol2csv.ofmvol2csv_main(
["ofm_example.vol"], "volfiles.csv", includefileorigin=False
)
first_frame = pd.read_csv("volfiles.csv")
sys.argv = ["csv2ofmvol", "volfiles.csv", "--output", "backagain.vol"]
csv2ofmvol.main()
ofmvol2csv.ofmvol2csv_main(["backagain.vol"], "take2.csv", includefileorigin=False)
second_frame = pd.read_csv("take2.csv")
print(first_frame.head())
print(second_frame.head())
pd.testing.assert_frame_equal(first_frame, second_frame)