OpenMC/tests/test_output/test_output.py
2013-02-01 15:22:32 -05:00

32 lines
824 B
Python

#!/usr/bin/env python
import os
from subprocess import Popen, STDOUT, PIPE
pwd = os.path.dirname(__file__)
def setup():
os.putenv('PWD', pwd)
os.chdir(pwd)
def test_run():
proc = Popen([pwd + '/../../src/openmc'], stderr=STDOUT, stdout=PIPE)
returncode = proc.wait()
print(proc.communicate()[0])
assert returncode == 0
def test_summary_exists():
assert os.path.exists(pwd + '/summary.out')
def test_cross_sections_exists():
assert os.path.exists(pwd + '/cross_sections.out')
def test_statepoint_exists():
assert os.path.exists(pwd + '/statepoint.10.binary')
def teardown():
output = [pwd + i for i in ['/statepoint.10.binary', '/summary.out',
'/cross_sections.out']]
for f in output:
if os.path.exists(f):
os.remove(f)