tests.py 454 B

1234567891011121314
  1. # Copyright (c) Sebastian Raschka under Apache License 2.0 (see LICENSE.txt).
  2. # Source for "Build a Large Language Model From Scratch"
  3. # - https://www.manning.com/books/build-a-large-language-model-from-scratch
  4. # Code: https://github.com/rasbt/LLMs-from-scratch
  5. # File for internal use (unit tests)
  6. from python_environment_check import main
  7. def test_main(capsys):
  8. main()
  9. captured = capsys.readouterr()
  10. assert "FAIL" not in captured.out