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

open Jest
open Expect
open TestHelpers
open FastCheck
open Arbitrary
open Property.Sync

describe("dotSubtract", () => {
  test("mean of normal minus exponential (unit)", () => {
    let mean = 0.0
    let rate = 10.0
    exception MeanFailed
    let dotDifference = DistributionOperation.Constructors.pointwiseSubtract(
      ~env,
      mkNormal(mean, 1.0),
      mkExponential(rate),
    )
    let meanResult = E.R2.bind(DistributionOperation.Constructors.mean(~env), dotDifference)
    let meanAnalytical =
      mean -.
      SymbolicDist.Exponential.mean({rate: rate})->E.R2.toExn(
        "On trusted input this should never happen",
      )
    switch meanResult {
    | Ok(meanValue) => meanValue->expect->toBeCloseTo(meanAnalytical)
    | Error(_) => raise(MeanFailed)
    }
  })
  /*
    It seems like this test should work, and it's plausible that
    there's some bug in `pointwiseSubtract`
 */
  Skip.test("mean of normal minus exponential (property)", () => {
    assert_(
      property2(
        float_(),
        floatRange(1e-5, 1e5),
        (mean, rate) => {
          // We limit ourselves to stdev=1 so that the integral is trivial
          let dotDifference = DistributionOperation.Constructors.pointwiseSubtract(
            ~env,
            mkNormal(mean, 1.0),
            mkExponential(rate),
          )
          let meanResult = E.R2.bind(DistributionOperation.Constructors.mean(~env), dotDifference)
          // according to algebra or random variables,
          let meanAnalytical =
            mean -.
            SymbolicDist.Exponential.mean({rate: rate})->E.R2.toExn(
              "On trusted input this should never happen",
            )
          switch meanResult {
          | Ok(meanValue) => abs_float(meanValue -. meanAnalytical) /. abs_float(meanValue) < 1e-2 // 1% relative error
          | Error(err) => err === DistributionTypes.OperationError(DivisionByZeroError)
          }
        },
      ),
    )
    pass
  })
})