squiggle/packages/squiggle-lang/__tests__/Distributions/AlgebraicCombination_test.res

198 lines
7.5 KiB
Plaintext
Raw Normal View History

open Jest
open Expect
open TestHelpers
let {
2022-04-11 18:53:02 +00:00
normalDist5, // mean=5, stdev=2
normalDist10, // mean=10, stdev=2
normalDist20, // mean=20, stdev=2
normalDist, // mean=5; stdev=2
uniformDist, // low=9; high=10
betaDist, // alpha=2; beta=5
lognormalDist, // mu=0; sigma=1
cauchyDist, // local=1; scale=1
triangularDist, // low=1; medium=2; high=3;
exponentialDist, // rate=2
} = module(GenericDist_Fixtures)
let {
algebraicAdd,
algebraicMultiply,
algebraicDivide,
algebraicSubtract,
algebraicLogarithm,
algebraicPower
} = module(DistributionOperation.Constructors)
2022-04-11 18:53:02 +00:00
let algebraicAdd = algebraicAdd(~env)
let algebraicMultiply = algebraicMultiply(~env)
let algebraicDivide = algebraicDivide(~env)
let algebraicSubtract = algebraicSubtract(~env)
let algebraicLogarithm = algebraicLogarithm(~env)
let algebraicPower = algebraicPower(~env)
2022-04-11 18:53:02 +00:00
describe("Addition of distributions", () => {
describe("mean", () => {
test("normal(mean=5) + normal(mean=20)", () => {
normalDist5
-> algebraicAdd(normalDist20)
-> E.R2.fmap(GenericDist_Types.Constructors.UsingDists.mean)
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toExn
-> expect
-> toBe(Some(2.5e1))
})
})
2022-04-11 18:53:02 +00:00
describe("pdf", () => {
2022-04-11 19:53:04 +00:00
testAll("(normal(mean=5) + normal(mean=5)).pdf (imprecise)", list{8e0, 1e1, 1.2e1, 1.4e1}, x => {
2022-04-11 18:53:02 +00:00
let expected = normalDist10
-> Ok
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.pdf(d, x))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
let calculated = normalDist5
-> algebraicAdd(normalDist5)
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.pdf(d, x))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
switch expected { // not exactly happy with this
| None => false -> expect -> toBe(true)
| Some(x) => switch calculated {
| None => false -> expect -> toBe(true)
| Some(y) => x -> expect -> toBeSoCloseTo(y, ~digits=0)
}
}
})
2022-04-11 19:53:04 +00:00
test("(normal(mean=10) + normal(mean=10)).pdf(1.9e1)", () => {
let expected = normalDist20
-> Ok
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.pdf(d, 1.9e1))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
let calculated = normalDist10
-> algebraicAdd(normalDist10)
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.pdf(d, 1.9e1))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
switch expected { // not exactly happy with this
| None => false -> expect -> toBe(true)
| Some(x) => switch calculated {
| None => false -> expect -> toBe(true)
| Some(y) => x -> expect -> toBeSoCloseTo(y, ~digits=1)
}
}
})
2022-04-11 18:53:02 +00:00
})
describe("cdf", () => {
2022-04-11 19:53:04 +00:00
testAll("(normal(mean=5) + normal(mean=5)).cdf (imprecise)", list{6e0, 8e0, 1e1, 1.2e1}, x => {
2022-04-11 18:53:02 +00:00
let expected = normalDist10
-> Ok
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.cdf(d, x))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
let calculated = normalDist5
-> algebraicAdd(normalDist5)
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.cdf(d, x))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
switch expected { // not exactly happy with this
| None => false -> expect -> toBe(true)
| Some(x) => switch calculated {
| None => false -> expect -> toBe(true)
| Some(y) => x -> expect -> toBeSoCloseTo(y, ~digits=0)
}
}
})
2022-04-11 19:53:04 +00:00
test("(normal(mean=10) + normal(mean=10)).cdf(1.25e1)", () => {
let expected = normalDist20
-> Ok
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.cdf(d, 1.25e1))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
let calculated = normalDist10
-> algebraicAdd(normalDist10)
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.cdf(d, 1.25e1))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
switch expected { // not exactly happy with this
| None => false -> expect -> toBe(true)
| Some(x) => switch calculated {
| None => false -> expect -> toBe(true)
| Some(y) => x -> expect -> toBeSoCloseTo(y, ~digits=2)
}
}
})
2022-04-11 18:53:02 +00:00
})
describe("inv", () => {
2022-04-11 19:53:04 +00:00
testAll("(normal(mean=5) + normal(mean=5)).inv (imprecise)", list{5e-2, 4.2e-3, 9e-3}, x => {
2022-04-11 18:53:02 +00:00
let expected = normalDist10
-> Ok
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.inv(d, x))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
let calculated = normalDist5
-> algebraicAdd(normalDist5)
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.inv(d, x))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
switch expected { // not exactly happy with this.
| None => false -> expect -> toBe(true)
| Some(x) => switch calculated {
| None => false -> expect -> toBe(true)
| Some(y) => x -> expect -> toBeSoCloseTo(y, ~digits=-1)
}
}
})
2022-04-11 19:53:04 +00:00
test("(normal(mean=10) + normal(mean=10)).inv(1e-1)", () => {
let expected = normalDist20
-> Ok
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.inv(d, 1e-1))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
let calculated = normalDist10
-> algebraicAdd(normalDist10)
-> E.R2.fmap(d => GenericDist_Types.Constructors.UsingDists.inv(d, 1e-1))
-> E.R2.fmap(run)
-> E.R2.fmap(toFloat)
-> E.R.toOption
-> E.O.flatten
switch expected { // not exactly happy with this
| None => false -> expect -> toBe(true)
| Some(x) => switch calculated {
| None => false -> expect -> toBe(true)
| Some(y) => x -> expect -> toBeSoCloseTo(y, ~digits=-1)
}
}
})
2022-04-11 18:53:02 +00:00
})
})