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

open Jest
open TestHelpers

describe("Combining Continuous and Discrete Distributions", () => {
  makeTest(
    "keep order of xs when multiplying by negative number",
    AlgebraicShapeCombination.isOrdered(
      AlgebraicShapeCombination.combineShapesContinuousDiscrete(
        #Multiply,
        {xs: [0., 1.], ys: [1., 1.]},
        {xs: [-1.], ys: [1.]},
        ~discretePosition=Second,
      ),
    ), // Multiply distribution by -1
    true,
  )
  makeTest(
    "keep order of xs when first number is discrete and adding",
    AlgebraicShapeCombination.isOrdered(
      AlgebraicShapeCombination.combineShapesContinuousDiscrete(
        #Add,
        {xs: [0., 1.], ys: [1., 1.]},
        {xs: [1.], ys: [1.]},
        ~discretePosition=First,
      ),
    ), // 1 + distribution
    true,
  )
})