tests/test_demo_api.py

from backend.universe import UniverseManager


def test_universe_manager():
    m = UniverseManager()
    uid = m.create_universe(["all odd numbers are integers", "prime numbers have exactly two divisors"])
    assert uid in m.universes

    res = m.prove(uid, "odd numbers")
    assert "status" in res