import torch

import beignet


def test_probabilists_hermite_polynomial_x():
    torch.testing.assert_close(
        beignet.probabilists_hermite_polynomial_x,
        torch.tensor([0.0, 1.0]),
        check_dtype=False,
    )
