Function test_std_must_be_non_zero

dali/test/python/torchvision/test_tv_normalize.py:154–167

(mean, std)

    ([0.0, 0.0, 0.0], [0.0, 1.0, 1.0]),
)
def test_std_must_be_non_zero(mean, std):
    x = make_sample_tensor()
    norm = transforms.Normalize(mean=mean, std=std)
    with assert_raises(ValueError):
        _ = norm(x)

    with assert_raises(ValueError):
        _ = Compose([Normalize(mean=mean, std=std)])

    with assert_raises(ValueError):
        _ = normalize(x, mean, std)

    with assert_raises(ValueError):
        _ = transforms.functional.normalize(x, mean, std)


"""