squiggle/packages/squiggle-lang/__tests__/Parser/Squiggle_test.res
2022-04-21 22:54:25 -04:00

36 lines
931 B
Plaintext

open Jest
open Expect
let {eval} = module(Parser_Squiggle)
describe("expressions of normal distributions:", () => {
test("sum of two", () => {
expect(eval(" normal (5 , 2 ) + normal(0,2)"))->toEqual(
{mean: 5.0 +. 0.0, stdev: Js.Math.sqrt(2.0 ** 2.0 +. 2.0 ** 2.0)}
->#Normal
->Symbolic
->EvDistribution
->Ok
->Some,
)
})
test("difference of two", () => {
expect(eval("normal(5,3)-normal(2,1)"))->toEqual(
{mean: 5.0 -. 2.0, stdev: Js.Math.sqrt(3.0 ** 2.0 +. 1.0 ** 2.0)}
->#Normal
->Symbolic
->EvDistribution
->Ok
->Some,
)
})
test("product of two", () => {
let item = eval("normal ( 5 , 3 ) * normal(2,1)")
exception NotOk(string)
let item' = switch item {
| Some(Ok(EvDistribution(SampleSet(x)))) => x
| _ => raise(NotOk("something happened"))
}
expect(Js.Array.length(item'))->toEqual(10000)
})
})