open Jest open Expect let env: GenericDist.env = { sampleCount: 100, xyPointLength: 100, } let { normalDist5, normalDist10, normalDist20, normalDist, uniformDist, betaDist, lognormalDist, cauchyDist, triangularDist, exponentialDist, } = module(GenericDist_Fixtures) let {toFloat, toDist, toString, toError, fmap} = module(DistributionOperation.Output) let {run} = module(DistributionOperation) let run = run(~env) let outputMap = fmap(~env) let toExt: option<'a> => 'a = E.O.toExt( "Should be impossible to reach (This error is in test file)", ) describe("sparkline", () => { let runTest = ( name: string, dist: DistributionTypes.genericDist, expected: DistributionOperation.outputType, ) => { test(name, () => { let result = DistributionOperation.run(~env, FromDist(#ToString(ToSparkline(20)), dist)) expect(result)->toEqual(expected) }) } runTest( "normal", normalDist, String(`▁▁▁▁▁▂▄▆▇██▇▆▄▂▁▁▁▁▁`), ) runTest( "uniform", uniformDist, String(`████████████████████`), ) runTest("beta", betaDist, String(`▁▄▇████▇▆▅▄▃▃▂▁▁▁▁▁▁`)) runTest( "lognormal", lognormalDist, String(`▁█▂▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁`), ) runTest( "cauchy", cauchyDist, String(`▁▁▁▁▁▁▁▁▁██▁▁▁▁▁▁▁▁▁`), ) runTest( "triangular", triangularDist, String(`▁▁▂▃▄▅▆▇████▇▆▅▄▃▂▁▁`), ) runTest( "exponential", exponentialDist, String(`█▅▄▂▂▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁`), ) }) describe("toPointSet", () => { test("on symbolic normal distribution", () => { let result = run(FromDist(#ToDist(ToPointSet), normalDist5)) ->outputMap(FromDist(#ToFloat(#Mean))) ->toFloat ->toExt expect(result)->toBeSoCloseTo(5.0, ~digits=0) }) test("on sample set", () => { let result = run(FromDist(#ToDist(ToPointSet), normalDist5)) ->outputMap(FromDist(#ToDist(ToSampleSet(1000)))) ->outputMap(FromDist(#ToDist(ToPointSet))) ->outputMap(FromDist(#ToFloat(#Mean))) ->toFloat ->toExt expect(result)->toBeSoCloseTo(5.0, ~digits=-1) }) })