diff --git a/integration/src/main/java/ai/djl/integration/tests/ndarray/NDArrayCreationOpTest.java b/integration/src/main/java/ai/djl/integration/tests/ndarray/NDArrayCreationOpTest.java index 3728410d92e..71cf7c816ce 100644 --- a/integration/src/main/java/ai/djl/integration/tests/ndarray/NDArrayCreationOpTest.java +++ b/integration/src/main/java/ai/djl/integration/tests/ndarray/NDArrayCreationOpTest.java @@ -413,6 +413,16 @@ public void testRandomNormal() { } } + @Test + public void testTruncatedNormal() { + try (NDManager manager = NDManager.newBaseManager()) { + NDArray normal = manager.truncatedNormal(new Shape(1000, 1000)); + Assertions.assertAlmostEquals(normal.mean().getFloat(), 0f, 2e-2f, 2e-2f); + Assert.assertTrue(normal.gte(-2).all().getBoolean()); + Assert.assertTrue(normal.lte(2).all().getBoolean()); + } + } + @Test public void testFixedSeed() { try (NDManager manager = NDManager.newBaseManager()) {