""" Code to profile on large mcnp files. Not using pytest. """ from zipfile import ZipFile from mckit.utils.resource import path_resolver from mckit.constants import MCNP_ENCODING from mckit import read_mcnp_text, Universe from mckit.parser.mcnp_input_sly_parser import ParseResult, from_text data_filename_resolver = path_resolver('benchmarks') with ZipFile(data_filename_resolver("data/4M.zip")) as data_archive: CLITE_TEXT = data_archive.read("clite.i").decode(encoding=MCNP_ENCODING) def test_old_mcnp_reading(): universe: Universe = read_mcnp_text(CLITE_TEXT) assert len(universe) == 150 def test_sly_mcnp_reading(): result: ParseResult = from_text(CLITE_TEXT) assert result.title == "C-LITE VERSION 1 RELEASE 131031 ISSUED 31/10/2013 - Halloween edition" assert len(result.universe) == 150 if __name__ == "__main__": test_old_mcnp_reading() test_sly_mcnp_reading()
def test_path_resolver_in_own_package_with_separate_file(): resolver = path_resolver() assert resolver("__init__.py").exists( ), "Should find __init__.py in the current package"
""" Code to profile on large mcnp files. Not using pytest. """ from zipfile import ZipFile from mckit import Universe from mckit.constants import MCNP_ENCODING from mckit.parser.mcnp_input_parser import read_mcnp_text from mckit.parser.mcnp_input_sly_parser import ParseResult, from_text from mckit.utils.resource import path_resolver data_filename_resolver = path_resolver("benchmarks") with ZipFile(data_filename_resolver("data/4M.zip")) as data_archive: CLITE_TEXT = data_archive.read("clite.i").decode(encoding=MCNP_ENCODING) def test_old_mcnp_reading(): universe: Universe = read_mcnp_text(CLITE_TEXT) assert len(universe) == 150 def test_sly_mcnp_reading(): result: ParseResult = from_text(CLITE_TEXT) assert ( result.title == "C-LITE VERSION 1 RELEASE 131031 ISSUED 31/10/2013 - Halloween edition" ) assert len(result.universe) == 150
def test_path_resolver(): resolver = path_resolver() actual = resolver(THIS_FILENAME) assert isinstance(actual, Path) assert actual.name == THIS_FILENAME