pytest.ini

# TF & some other libraries report a bunch of deprecation warnings
[pytest]

# Ignore specific tests
addopts = -svv

# Add pytest markers
markers =
    push: marks tests as push
    nightly: marks tests as nightly

# Where pytest should look for tests
testpaths =
    # Ops
    forge/test/mlir/test_ops.py
    forge/test/mlir/test_ops_tf.py

    # Features
    forge/test/mlir/test_features.py

    # API
    forge/test/test_api.py

    # Model Tests (CNN)
    forge/test/model_demos/high_prio/cnn/pytorch

    # Model Tests (NLP)
    forge/test/model_demos/high_prio/nlp/pytorch

    # MNIST Linear
    forge/test/mlir/mnist/test_inference.py
    forge/test/mlir/mnist/training/test_training.py
    forge/test/mlir/test_training.py

    # Llama
    forge/test/mlir/llama/test_llama_inference.py::test_llama_inference
    forge/test/mlir/llama/tests

    # Resnet
    forge/test/mlir/resnet/test_resnet_inference.py::test_resnet_inference

    # Benchmark
    # MNIST Linear
    forge/test/benchmark/benchmark/models/mnist_linear.py::test_mnist_linear

filterwarnings =
    ignore::DeprecationWarning