Skip to content

argumentcomputer/LSpec

Repository files navigation

LSpec

A testing framework for Lean 4, inspired by Haskell's Hspec package.

Usage

Composing tests

Sequences of tests are represented by the TestSeq datatype. In order to instantiate terms of TestSeq, use the test helper function:

#check
  test "Nat equality" (4 = 4) $
  test "Nat inequality" (45)
-- test "Nat equality" (4 = 4) (test "Nat inequality" (4 ≠ 5)) : TestSeq

test consumes a description a proposition and a next test The proposition, however, must have its own instance of Testable.

You can also collect TestSeq into conceptual test groups by using the helper function group:

#check
  test "Nat equality" (42 = 42) $
  group "manual group" $
    test "Nat equality inside group" (4 = 4)

The Testable class

Testable is how Lean is instructed to decide whether certain propositions are resolved as true or false.

This is an example of a simple instance for decidability of equalities:

instance (x y : α) [DecidableEq α] [Repr α] : Testable (x = y) :=
  if h : x = y then
    .isTrue h
  else
    .isFalse h s!"Not equal: {repr x} and {repr y}"

The custom failure message is optional.

There are more examples of Testable instances in LSpec/Instances.lean.

The user is, of course, free to provide their own instances.

Actually running the tests

The #lspec command

The #lspec command allows you to test interactively in a file.

Examples:

#lspec
  test "four equals four" (4 = 4) $
  test "five equals five" (5 = 5)
-- ✓ four equals four
-- ✓ five equals five

An important note is that a failing test will raise an error, interrupting the building process.

The lspecIO function

lspecIO is meant to be used in files to be compiled and integrated in a testing infrastructure, as shown below.

def aaSuite := [
    test "four equals four" (4 = 4)
  ]

def bbSuite := [
    test "five equals five" (5 = 5)
  ]

def main := lspecIO $ .ofList [
    ("aa", aaSuite),
    ("bb", bbSuite)
  ]

Once such main function is defined, its respective executable can be tagged as the @[test_driver] in the lakefile. For further information, inspect the docstring of lspecIO.

Integration with SlimCheck

There are 3 main typeclasses associated with any SlimCheck test:

  • Shrinkable : The typeclass that takes a type a : α and returns a List α of elements which should be thought of as being "smaller" than a (in some sense dependent on the type α being considered).
  • SampleableExt : The typeclass of a . This is roughly equivalent to QuickCheck's Arbitrary typeclass.
  • Checkable : The property to be checked by SlimCheck must have a Checkable instance.

In order to use SlimCheck tests for custom data types, the user will need to implement instances of the typeclasses Shrinkable and SampleableExt for the custom types appearing in the properties being tested.

The module LSpec.SlimCheck.Checkable contains may of the useful definitions and instances that can be used to derive a Checkable instance for a wide variety of properties given just the instances above. If all else fails, the user can also define the Checkable instance by hand.

Once this is done a Slimcheck test is evaluated in a similar way to LSpec tests:

#lspec check "add_comm" $ ∀ n m : Nat, n + m = m + n

#lspec check "add_comm" $ ∀ n m : Nat, n + m = m + m
-- × add_comm

-- ===================
-- Found problems!
-- n := 1
-- m := 0
-- issue: 1 = 0 does not hold
-- (0 shrinks)
-- -------------------

Integration with Plausible

LSpec also integrates with Lean's Plausible property-based testing library. The Plausible backend lives alongside the SlimCheck-based check/checkIO described above rather than replacing them, so existing SlimCheck tests continue to work unchanged.

Plausible relies on the same core typeclasses as QuickCheck — Shrinkable and SampleableExt to generate and shrink random values — plus Plausible.Testable for the property itself. Instances for the common types (Nat, Int, List, etc.) ship with Plausible, and custom types are supported by providing Shrinkable/SampleableExt instances just as with SlimCheck.

The module LSpec.Plausible exposes two macros:

  • checkPlausible' — a compile-time property test, evaluated during elaboration with a fixed random seed (deterministic across compilations). This is the Plausible-backed counterpart to check'.
  • checkPlausibleIO' — a runtime property test, deferred until the test suite is run. This enables fresh random values on each run and configurable seeds via cfg.randomSeed. This is the Plausible-backed counterpart to checkIO'.

Both macros capture the property syntax so it appears in the output. (Non-syntax-capturing checkPlausible/checkPlausibleIO functions are also available if you don't need the property echoed back.)

A compile-time test with #lspec:

#lspec checkPlausible' "add_comm" (∀ n m : Nat, n + m = m + n)
-- ✓ ∃₁₀₀: "add_comm" (∀ n m : Nat, n + m = m + n)

#lspec checkPlausible' "bad" (∀ n : Nat, n < 5)
-- × ∃¹⁰/₁₀₀: "bad" (∀ n : Nat, n < 5)

-- ===================
-- Found problems!
-- n := 6
-- issue: 6 < 5 does not hold
-- (0 shrinks)
-- -------------------

A runtime test, run via lspecIO. Because checkPlausibleIO' tests are skipped by the pure #lspec runner, they must be executed with lspecIO (or lspecEachIO):

open LSpec

def plausibleTests : TestSeq :=
  checkPlausibleIO' "add_comm" (∀ n m : Nat, n + m = m + n)

def main : IO UInt32 := lspecIO (.ofList [("plausibleTests", [plausibleTests])]) []

Multiple property tests can be sequenced with ++. Note that the '-suffixed macros capture everything up to the end of the line as the property, so to chain them use the non-capturing checkPlausibleIO function (which takes an explicit next argument):

def suite : TestSeq :=
  checkPlausibleIO "add_comm" (∀ n m : Nat, n + m = m + n) $
  checkPlausibleIO "mul_one"  (∀ n : Nat, n * 1 = n)

The '-suffixed macros always use the default configuration. To pass a fixed seed for reproducible runs (or otherwise customise the Plausible.Configuration), call the underlying checkPlausibleIO function directly:

def reproducible : TestSeq :=
  checkPlausibleIO "add_comm" (∀ n m : Nat, n + m = m + n) .done { randomSeed := some 42 }

About

A Testing Framework for Lean

Resources

Stars

Watchers

Forks

Releases

Packages

Used by

Contributors

Languages