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