squiggle/packages/squiggle-lang/__tests__/Reducer/Reducer_TestHelpers.res

module Expression = Reducer.Expression
module ExpressionValue = ReducerInterface.ExpressionValue
module ErrorValue = Reducer_ErrorValue

open Jest
open Expect

let unwrapRecord = rValue =>
  rValue->Belt.Result.flatMap(value =>
    switch value {
    | ExpressionValue.EvRecord(aRecord) => Ok(aRecord)
    | _ => ErrorValue.RETodo("TODO: External bindings must be returned")->Error
    }
  )

let expectParseToBe = (expr: string, answer: string) =>
  Reducer.parse(expr)->Expression.toStringResult->expect->toBe(answer)

let expectParseOuterToBe = (expr: string, answer: string) =>
  Reducer.parseOuter(expr)->Expression.toStringResult->expect->toBe(answer)

let expectParsePartialToBe = (expr: string, answer: string) =>
  Reducer.parsePartial(expr)->Expression.toStringResult->expect->toBe(answer)

let expectEvalToBe = (expr: string, answer: string) =>
  Reducer.evaluate(expr)->ExpressionValue.toStringResult->expect->toBe(answer)

let expectEvalBindingsToBe = (expr: string, bindings: Reducer.externalBindings, answer: string) =>
  Reducer.evaluateUsingOptions(
    expr,
    ~externalBindings=Some(bindings),
    ~isPartial=None,
    ~environment=None,
  )
  ->ExpressionValue.toStringResult
  ->expect
  ->toBe(answer)

let expectEvalPartialBindingsToBe = (
  expr: string,
  bindings: Reducer.externalBindings,
  answer: string,
) =>
  Reducer.evaluateUsingOptions(
    expr,
    ~externalBindings=Some(bindings),
    ~isPartial=Some(true),
    ~environment=None,
  )
  ->unwrapRecord
  ->ExpressionValue.toStringResultRecord
  ->expect
  ->toBe(answer)

let testParseToBe = (expr, answer) => test(expr, () => expectParseToBe(expr, answer))
let testParseOuterToBe = (expr, answer) => test(expr, () => expectParseOuterToBe(expr, answer))
let testParsePartialToBe = (expr, answer) => test(expr, () => expectParsePartialToBe(expr, answer))
let testDescriptionParseToBe = (desc, expr, answer) =>
  test(desc, () => expectParseToBe(expr, answer))

let testEvalToBe = (expr, answer) => test(expr, () => expectEvalToBe(expr, answer))
let testDescriptionEvalToBe = (desc, expr, answer) => test(desc, () => expectEvalToBe(expr, answer))
let testEvalBindingsToBe = (expr, bindingsList, answer) =>
  test(expr, () => expectEvalBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer))
let testEvalPartialBindingsToBe = (expr, bindingsList, answer) =>
  test(expr, () => expectEvalPartialBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer))

module MySkip = {
  let testParseToBe = (expr, answer) => Skip.test(expr, () => expectParseToBe(expr, answer))
  let testParseOuterToBe = (expr, answer) =>
    Skip.test(expr, () => expectParseOuterToBe(expr, answer))
  let testParsePartialToBe = (expr, answer) =>
    Skip.test(expr, () => expectParsePartialToBe(expr, answer))
  let testEvalToBe = (expr, answer) => Skip.test(expr, () => expectEvalToBe(expr, answer))
  let testEvalBindingsToBe = (expr, bindingsList, answer) =>
    Skip.test(expr, () => expectEvalBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer))
  let testEvalPartialBindingsToBe = (expr, bindingsList, answer) =>
    Skip.test(expr, () =>
      expectEvalPartialBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer)
    )
}
module MyOnly = {
  let testParseToBe = (expr, answer) => Only.test(expr, () => expectParseToBe(expr, answer))
  let testParseOuterToBe = (expr, answer) =>
    Only.test(expr, () => expectParseOuterToBe(expr, answer))
  let testParsePartialToBe = (expr, answer) =>
    Only.test(expr, () => expectParsePartialToBe(expr, answer))
  let testEvalToBe = (expr, answer) => Only.test(expr, () => expectEvalToBe(expr, answer))
  let testEvalBindingsToBe = (expr, bindingsList, answer) =>
    Only.test(expr, () => expectEvalBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer))
  let testEvalPartialBindingsToBe = (expr, bindingsList, answer) =>
    Only.test(expr, () =>
      expectEvalPartialBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer)
    )
}