Commit fabddb9c authored by Patrick Shriwise's avatar Patrick Shriwise
Browse files

Updating tests

parent 1d82195b

tests/test_material_report.py

+6 −5
Original line number Diff line number Diff line
@@ -6,7 +6,8 @@ from pygriffin.pygriffin import PyGriffin 


def test_isoxml_isotopes(request):

    exp_isotopes = ['pseudo_REG_A', 'pseudo_REG_B', 'pseudo_REG_C']

    isotopes = parse_isoxml(request.path.parent / 'test_files/isoxml_sample.xml')

    isotopes = parse_isoxml(request.path.parent / 'test_files/isoxml_sample.xml',

                            "ISOTXS-neutron")

    assert exp_isotopes == isotopes
@@ -16,7 +17,9 @@ def test_materials(request): 
    # create a pygriffin instance

    pyg = PyGriffin(input=griffin_input, mesh=griffin_mesh)



    mat_assignments = pyg.get_material_assignments()

    library, mat_assignments = pyg.get_material_assignments()



    assert library == "ISOTXS-neutron"



    exp_mat_assignments = {'fuel_assembly2_z0': ['pseudo_REG_B'],

                           'fuel_assembly1_z0': ['pseudo_REG_A'],
@@ -24,7 +27,5 @@ def test_materials(request): 
    assert sorted(exp_mat_assignments.keys()) == sorted(mat_assignments.keys())



    exp_isotopes = (['pseudo_REG_B'], ['pseudo_REG_A'], ['pseudo_REG_C'])

    for exp_isos, mat_isos in zip(exp_isotopes, mat_assignments.values()):

    for exp_isos, (mat_isos, _) in zip(exp_isotopes, mat_assignments.values()):

        assert exp_isos == mat_isos