def test_conformance2(): """Make sure the file is removed first""" fn = DEFAULT_PARAMS["filename"] if os.path.exists(fn): os.remove(fn) tag = "some magic string to find" with open(fn, "w") as f: f.write(tag) lattice_file.write(**DEFAULT_PARAMS) with open(fn) as f: res = f.read() assert not re.search(tag, res), "file was not removed" assert re.search("VERSION", res)
def test_conformance1(): """Read the file and make sure it is formated correctly""" fn = DEFAULT_PARAMS["filename"] if os.path.exists(fn): os.remove(fn) lattice_file.write(**DEFAULT_PARAMS) with open(fn) as f: res = f.read() def a(p, flags=re.MULTILINE): assert re.search(p, res, flags=flags), p + ": no match" a(r"--\nQF 1\.1 1\.2 1\.3 \n") a(r" \nAW 5\.1 5\.2 5\.3 \n$", flags=0) a(r"QF +1.1 .*\nQF +2.1 .*\n(?:#.*\n)+AW +3.1 .*\nAW +4.1 .*\nAW +5.1 ")