Simplify test

This commit is contained in:
Alexander Nozik 2021-04-10 17:15:15 +03:00
parent 4abd0bdb6f
commit 000c79d42f

View File

@ -1,24 +1,21 @@
package space.kscience.kmath.stat
import kotlinx.coroutines.flow.take
import kotlinx.coroutines.flow.toList
import kotlinx.coroutines.runBlocking
import org.junit.jupiter.api.Assertions
import org.junit.jupiter.api.Test
import space.kscience.kmath.samplers.GaussianSampler
import space.kscience.kmath.structures.asBuffer
internal class CommonsDistributionsTest {
@Test
fun testNormalDistributionSuspend() = runBlocking {
val distribution = GaussianSampler(7.0, 2.0)
val generator = RandomGenerator.default(1)
val sample = distribution.sample(generator).take(1000).toList().toDoubleArray().asBuffer()
val sample = distribution.sample(generator).nextBuffer(1000)
Assertions.assertEquals(7.0, Mean.double(sample), 0.2)
}
@Test
fun testNormalDistributionBlocking() {
fun testNormalDistributionBlocking() {
val distribution = GaussianSampler(7.0, 2.0)
val generator = RandomGenerator.default(1)
val sample = distribution.sample(generator).nextBufferBlocking(1000)