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