Python apply_thresholds Examples

Example #1

def test_apply_thresholds_with_variable_threshold():
    input = numpy.array([1000, 1000, 1000])
    thresholds = [numpy.array([500, 1500, 1000])
                  ]  # Only one thresold, but varies with the person
    choice_list = [True, False]  # True if input <= threshold, false otherwise
    result = apply_thresholds(input, thresholds, choice_list)
    assert_near(result, [False, True, True])

Example #2

def test_apply_thresholds_with_too_many_thresholds():
    input = numpy.array([10])
    thresholds = [5, 4]
    choice_list = [10]
    with pytest.raises(AssertionError):
        return apply_thresholds(input, thresholds, choice_list)

Example #3

def test_apply_thresholds_with_as_many_thresholds_than_choices():
    input = numpy.array([4, 6, 8])
    thresholds = [5, 7]
    choice_list = [10, 20]
    result = apply_thresholds(input, thresholds, choice_list)
    assert_near(result, [10, 20, 0])

Example #4

File: test_formula_helpers.py Project: willi-am-publi/openfisca-core

def test_apply_thresholds_with_too_few_thresholds():
    input = numpy.array([10])
    thresholds = [5]
    choice_list = [10, 15, 20]
    return apply_thresholds(input, thresholds, choice_list)

Example #5

def test_apply_thresholds():
    input = numpy.array([4, 5, 6, 7, 8])
    thresholds = [5, 7]
    choice_list = [10, 15, 20]
    result = apply_thresholds(input, thresholds, choice_list)
    assert_near(result, [10, 10, 15, 15, 20])

Example #6

File: test_formula_helpers.py Project: ACPU/openfisca-core

def test_apply_thresholds_with_variable_threshold():
    input = numpy.array([1000, 1000, 1000])
    thresholds = [numpy.array([500, 1500, 1000])]  # Only one thresold, but varies with the person
    choice_list = [True, False]  # True if input <= threshold, false otherwise
    result = apply_thresholds(input, thresholds, choice_list)
    assert_near(result, [False, True, True])

Example #7

File: test_formula_helpers.py Project: ACPU/openfisca-core

def test_apply_thresholds_with_as_many_thresholds_than_choices():
    input = numpy.array([4, 6, 8])
    thresholds = [5, 7]
    choice_list = [10, 20]
    result = apply_thresholds(input, thresholds, choice_list)
    assert_near(result, [10, 20, 0])

Example #8

File: test_formula_helpers.py Project: ACPU/openfisca-core

def test_apply_thresholds():
    input = numpy.array([4, 5, 6, 7, 8])
    thresholds = [5, 7]
    choice_list = [10, 15, 20]
    result = apply_thresholds(input, thresholds, choice_list)
    assert_near(result, [10, 10, 15, 15, 20])

Example #9

File: test_formula_helpers.py Project: ACPU/openfisca-core

def test_apply_thresholds_with_too_few_thresholds():
    input = numpy.array([10])
    thresholds = [5]
    choice_list = [10, 15, 20]
    return apply_thresholds(input, thresholds, choice_list)

Example #10

File: test_formula_helpers.py Project: openfisca/openfisca-core

def test_apply_thresholds_with_too_many_thresholds():
    input = numpy.array([10])
    thresholds = [5, 4]
    choice_list = [10]
    with pytest.raises(AssertionError):
        return apply_thresholds(input, thresholds, choice_list)