Add function name pattern for pytest and align tests

Author: Victor-Jung

Tests/test_mnist.py

@@ -131,7 +131,7 @@ def calibrate_model(
             model(images)
 
 
-def test_mnist_quant_export() -> None:
+def deepQuantTestMnistQuantExport() -> None:
     """Main execution function."""
 
     DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")

Tests/test_resnet18.py

@@ -115,7 +115,7 @@ def prepare_resnet18_model() -> nn.Module:
     return quantized_resnet
 
 
-def test_resnet18_quant_export() -> None:
+def deepQuantTestResnet18QuantExport() -> None:
     """
     Test function for exporting a quantized ResNet18 using BrevitasExporter.
     Validates the export process by running exportBrevitas and printing the FX graph.

Tests/test_simple_cnn.py

@@ -112,7 +112,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
         return x
 
 
-def test_simple_quant_cnn() -> None:
+def deepQuantTestSimpleQuantCNN() -> None:
     """
     Test function for the SimpleQuantCNN using exportBrevitas.
     Ensures both the forward pass and the FX tracing work properly.

Tests/test_simple_mha.py

@@ -72,7 +72,7 @@ def forward(self, x: Tensor) -> Tensor:
         return out
 
 
-def test_simple_quant_mha() -> None:
+def deepQuantTestSimpleQuantMHA() -> None:
     """
     Test function for the SimpleQuantMHA using exportBrevitas.
     Verifies both the forward pass and the export tracing.

Tests/test_simple_nn.py

@@ -94,7 +94,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
         return x
 
 
-def test_simple_quant_nn() -> None:
+def deepQuantTestSimpleQuantNN() -> None:
     """
     Test function for the SimpleQuantNN using exportBrevitas.
     Tests both the model's functionality and the export process.

pyproject.toml

@@ -39,6 +39,7 @@ dependencies = [
 packages = ["DeepQuant"]
 
 [tool.pytest.ini_options]
+python_functions = ["deepQuantTest*"]
 filterwarnings = [
     "ignore::DeprecationWarning",
     "ignore::UserWarning"