Python get_args Examples

Example #1

File: test_eddy.py Project: Shimwell/Eddy-Source

def test_get_args_with_bad_passed_sf(mocker):
    # arrange
    name = 'mcnp_examples/F2.out'
    sf = 'parrot'
    mocker.patch(
        'eddymc.eddy.argparse.ArgumentParser.parse_args',
        return_value=Namespace(file=None, scaling_factor=None),
    )
    # act, assert
    with pytest.raises(ValueError):
        eddy.get_args(name, sf)

Example #2

File: test_eddy.py Project: Shimwell/Eddy-Source

def test_get_args_with_passed_arguments(mocker, f2_file):
    # arrange
    name = 'mcnp_examples/F2.out'
    data = f2_file
    sf = 3.141592
    mocker.patch(
        'eddymc.eddy.argparse.ArgumentParser.parse_args',
        return_value=Namespace(file=None, scaling_factor=None),
    )
    # act
    name, output_data, sf, crit = eddy.get_args(name, sf)
    # assert
    assert name == 'mcnp_examples/F2.out'
    assert output_data == data
    assert sf == 3.141592
    assert crit is False

Example #3

File: test_eddy.py Project: Cerberus-Nuclear/Eddy-Source

def test_get_args_with_passed_arguments(mocker):
    # arrange
    name = 'mcnp_examples/F2.out'
    with open('mcnp_examples/F2.out', 'r') as file:
        data = file.readlines()
    sf = 3.141592
    mocker.patch(
        'eddymc.eddy.argparse.ArgumentParser.parse_args',
        return_value=Namespace(file=None, scaling_factor=None),
    )
    # act
    name, output_data, sf, crit = eddy.get_args(name, sf)
    # assert
    assert name == 'mcnp_examples/F2.out'
    #assert output_data == data  # no longer works since input is sterilized
    assert sf == 3.141592
    assert crit is False