コード例 #1
0
ファイル: test_nepc.py プロジェクト: padamson/nepc
def test_CS_class(nepc_connect):
    """Verify that the nepc.CS class has metadata and data attributes,
    and that each attribute is of the correct type"""
    # FIXME: randomly sample a few cross sections in the database
    cs = nepc.CS(nepc_connect[1], 1)
    # FIXME: add assert for each key in metadata
    assert isinstance(cs.metadata, dict)
    assert isinstance(cs.data, dict)
    assert isinstance(cs.metadata["cs_id"], int)
    assert isinstance(cs.metadata["units_e"], float)
    assert isinstance(cs.data["e"], list)
    assert isinstance(cs.data["e"][0], float)
    assert isinstance(cs.data["sigma"], list)
    assert isinstance(cs.data["sigma"][0], float)
    # test len()
    assert len(cs) == 21
    # test str()
    message = "cross section id: 1\n" \
            + "process: excitation\n" \
            + "reaction: E + N2(X1Sigmag+) -> E + N2(X1Sigmag+)_jSCHULZ\n" \
            + "threshold: 0.02 eV\n" \
            + "ref: N/A\n" \
            + "background: SCHULZ Lorem ipsum dolor sit amet, " \
            + "consectetur adipiscing elit, sed do eiusmod tempor " \
            + "incididunt ut labore et dolore magna aliqua. " \
            + "Ut enim ad minim veniam, quis nostrud exercitation ullamco " \
            + "laboris nisi ut aliquip ex ea commodo consequat. " \
            + "Duis aute irure dolor in reprehenderit in voluptate " \
            + "velit esse cillum dolore eu fugiat nulla pariatur.\n"
    assert(str(cs) == message)
コード例 #2
0
ファイル: test_nepc.py プロジェクト: ndisner/nepc
def test_reaction_latex(nepc_connect):
    """Verify when nepc.reaction_latex is called it
    returns a string representing the LaTeX
    for the reaction from a nepc cross section"""
    # FIXME: verify latex is correct
    # FIXME: randomly sample cross sections
    for i in range(1, 30):
        cs = nepc.CS(nepc_connect[1], i)
        assert isinstance(nepc.reaction_latex(cs), str)
コード例 #3
0
ファイル: test_nepc.py プロジェクト: ndisner/nepc
def test_reaction_text(nepc_connect):
    """Verify nepc.reaction_text and nepc.reaction_text_side
    return strings representing the LHS, RHS, or full plain text
    for the reaction from a nepc cross section"""
    # TODO: sample all process types and enough permutations
    cs = nepc.CS(nepc_connect[1], 1)
    lhsA = 'N2(X1Sigmag+)'
    rhsA = 'N2(X1Sigmag+)_jSCHULZ'
    assert nepc.reaction_text_side('LHS', cs) == (lhsA, f'E + {lhsA}')
    assert nepc.reaction_text_side('RHS', cs) == (rhsA, f'E + {rhsA}')
    assert nepc.reaction_text(cs) == (f'{lhsA} -> {rhsA}',
                                      f'E + {lhsA} -> E + {rhsA}')
コード例 #4
0
ファイル: test_nepc.py プロジェクト: ndisner/nepc
def test_CS_class(nepc_connect):
    """Verify that the nepc.CS class has metadata and data attributes,
    and that each attribute is of the correct type"""
    # FIXME: randomly sample a few cross sections in the database
    cs = nepc.CS(nepc_connect[1], 1)
    # FIXME: add assert for each key in metadata
    assert isinstance(cs.metadata, dict)
    assert isinstance(cs.data, dict)
    assert isinstance(cs.metadata["cs_id"], int)
    assert isinstance(cs.metadata["specie"], str)
    assert isinstance(cs.metadata["units_e"], float)
    assert isinstance(cs.data["e"], list)
    assert isinstance(cs.data["e"][0], float)
    assert isinstance(cs.data["sigma"], list)
    assert isinstance(cs.data["sigma"][0], float)
    # test len()
    # FIXME: use length of data file to get length of sampled cross section
    assert len(cs) == 21
コード例 #5
0
ファイル: test_nepc.py プロジェクト: padamson/nepc
def test_reaction_text(nepc_connect):
    """Verify nepc.reaction_text and nepc.reaction_text_side
    return strings representing the LHS, RHS, or full plain text
    for the reaction from a nepc cross section"""
    # TODO: sample all process types and enough permutations
    for i in range(1, 30):
        cs = nepc.CS(nepc_connect[1], i)
        lhsA = cs.metadata['lhsA']
        rhsA = cs.metadata['rhsA']
        if cs.metadata['process'] == 'excitation_v':
            lhsA = lhsA.replace(")", " v=" + str(cs.metadata['lhs_v']) + ")")
            rhsA = rhsA.replace(")", " v=" + str(cs.metadata['rhs_v']) + ")")
        assert isinstance(cs.reaction_latex, str)
        assert cs.reaction_text_side('LHS') \
                == (lhsA, f"{cs.metadata['e_on_lhs']if cs.metadata['e_on_lhs'] > 1 else ''}E + {lhsA}")
        assert cs.reaction_text_side('RHS') \
                == (rhsA, f"{cs.metadata['e_on_rhs']if cs.metadata['e_on_rhs'] > 1 else ''}E + {rhsA}")
        assert cs.reaction_text == (f"{lhsA} -> {rhsA}", 
                                      f"{cs.metadata['e_on_lhs']if cs.metadata['e_on_lhs'] > 1 else ''}" + \
                                      f"E + {lhsA} -> " + \
                                      f"{cs.metadata['e_on_rhs']if cs.metadata['e_on_rhs'] > 1 else ''}" + \
                                      f"E + {rhsA}")
コード例 #6
0
ファイル: test_nepc.py プロジェクト: padamson/nepc
def test_reaction_latex(nepc_connect):
    """Verify when nepc.reaction_latex is called it
    returns a string representing the LaTeX
    for the reaction from a nepc cross section"""
    # FIXME: verify latex is correct
    # FIXME: randomly sample cross sections
    for i in range(1, 30):
        cs = nepc.CS(nepc_connect[1], i)
        lhsA_long = cs.metadata['lhsA_long']
        rhsA_long = cs.metadata['rhsA_long']
        if cs.metadata['process'] == 'excitation_v':
            lhsA_long = lhsA_long.replace(")", " v=" + str(cs.metadata['lhs_v']) + ")")
            rhsA_long = rhsA_long.replace(")", " v=" + str(cs.metadata['rhs_v']) + ")")
        assert isinstance(cs.reaction_latex, str)
        assert cs.reaction_text_side('LHS', latex=True) \
                == f"{cs.metadata['e_on_lhs']if cs.metadata['e_on_lhs'] > 1 else ''}e$^-$ + {lhsA_long}"
        assert cs.reaction_text_side('RHS', latex=True) \
                == f"{cs.metadata['e_on_rhs']if cs.metadata['e_on_rhs'] > 1 else ''}e$^-$ + {rhsA_long}"
        assert cs.reaction_latex == f"{cs.metadata['e_on_lhs']if cs.metadata['e_on_lhs'] > 1 else ''}" + \
                                      f"e$^-$ + {lhsA_long} $\\rightarrow$ " + \
                                      f"{cs.metadata['e_on_rhs']if cs.metadata['e_on_rhs'] > 1 else ''}" + \
                                      f"e$^-$ + {rhsA_long}"