Example #1
0
def test_perturb_type_check():
    with pytest.raises(ValueError):
        perturb(0)
Example #2
0
def test_perturb(x, result):
    assert perturb(x) == result  # Test NumPy.
    assert perturb(tf.constant(x)).numpy() == result  # Test TF.