From 9b107b928b60903037aa1baf5d82f0d6047a3705 Mon Sep 17 00:00:00 2001
From: GALI PREM SAGAR <sagarprem75@gmail.com>
Date: Mon, 30 Sep 2024 14:28:42 -0500
Subject: [PATCH] Swtch traceback to `--native` in `cugraph` (#4663)

In cudf we have observed a ~10% speed up of pytest suite execution by switching pytest traceback to `--native`:

```
currently:

102474 passed, 2117 skipped, 902 xfailed in 892.16s (0:14:52)

--tb=short:

102474 passed, 2117 skipped, 902 xfailed in 898.99s (0:14:58)

--tb=no:

102474 passed, 2117 skipped, 902 xfailed in 815.98s (0:13:35)

--tb=native:

102474 passed, 2117 skipped, 902 xfailed in 820.92s (0:13:40)
```

This PR makes similar change to `cugraph` repo.

xref: https://github.com/rapidsai/cudf/pull/16851

Authors:
  - GALI PREM SAGAR (https://github.com/galipremsagar)

Approvers:
  - Brad Rees (https://github.com/BradReesWork)

URL: https://github.com/rapidsai/cugraph/pull/4663
---
 benchmarks/pytest.ini                                         | 1 +
 .../cugraph-equivariant/cugraph_equivariant/tests/pytest.ini  | 4 ++++
 python/cugraph-pyg/cugraph_pyg/tests/pytest.ini               | 4 ++++
 python/cugraph-pyg/pytest.ini                                 | 1 +
 python/cugraph-service/pytest.ini                             | 1 +
 python/cugraph-service/tests/pytest.ini                       | 4 ++++
 python/cugraph/pytest.ini                                     | 1 +
 python/nx-cugraph/nx_cugraph/tests/pytest.ini                 | 4 ++++
 python/pylibcugraph/pytest.ini                                | 2 ++
 9 files changed, 22 insertions(+)
 create mode 100644 python/cugraph-equivariant/cugraph_equivariant/tests/pytest.ini
 create mode 100644 python/cugraph-pyg/cugraph_pyg/tests/pytest.ini
 create mode 100644 python/cugraph-service/tests/pytest.ini
 create mode 100644 python/nx-cugraph/nx_cugraph/tests/pytest.ini

diff --git a/benchmarks/pytest.ini b/benchmarks/pytest.ini
index fe7fc31b6d6..d692b78de37 100644
--- a/benchmarks/pytest.ini
+++ b/benchmarks/pytest.ini
@@ -8,6 +8,7 @@ testpaths =
 
 addopts =
     --benchmark-columns="min, max, mean, stddev, outliers"
+    --tb=native
 
 markers =
     managedmem_on: RMM managed memory enabled
diff --git a/python/cugraph-equivariant/cugraph_equivariant/tests/pytest.ini b/python/cugraph-equivariant/cugraph_equivariant/tests/pytest.ini
new file mode 100644
index 00000000000..7b0a9f29fb1
--- /dev/null
+++ b/python/cugraph-equivariant/cugraph_equivariant/tests/pytest.ini
@@ -0,0 +1,4 @@
+# Copyright (c) 2024, NVIDIA CORPORATION.
+
+[pytest]
+addopts = --tb=native
diff --git a/python/cugraph-pyg/cugraph_pyg/tests/pytest.ini b/python/cugraph-pyg/cugraph_pyg/tests/pytest.ini
new file mode 100644
index 00000000000..7b0a9f29fb1
--- /dev/null
+++ b/python/cugraph-pyg/cugraph_pyg/tests/pytest.ini
@@ -0,0 +1,4 @@
+# Copyright (c) 2024, NVIDIA CORPORATION.
+
+[pytest]
+addopts = --tb=native
diff --git a/python/cugraph-pyg/pytest.ini b/python/cugraph-pyg/pytest.ini
index db99a54ae49..07c4ffa0958 100644
--- a/python/cugraph-pyg/pytest.ini
+++ b/python/cugraph-pyg/pytest.ini
@@ -17,6 +17,7 @@ addopts =
            --benchmark-max-time=0
            --benchmark-min-rounds=1
            --benchmark-columns="mean, rounds"
+           --tb=native
            ## do not run slow tests/benchmarks by default
            -m "not slow"
 
diff --git a/python/cugraph-service/pytest.ini b/python/cugraph-service/pytest.ini
index 6a0dd36ecec..f2ba9175f82 100644
--- a/python/cugraph-service/pytest.ini
+++ b/python/cugraph-service/pytest.ini
@@ -16,6 +16,7 @@ addopts = --benchmark-warmup=off
           --benchmark-max-time=0
           --benchmark-min-rounds=1
           --benchmark-columns="min, max, mean, rounds"
+          --tb=native
           ## for use with rapids-pytest-benchmark plugin
           #--benchmark-gpu-disable
           ## for use with pytest-cov plugin
diff --git a/python/cugraph-service/tests/pytest.ini b/python/cugraph-service/tests/pytest.ini
new file mode 100644
index 00000000000..7b0a9f29fb1
--- /dev/null
+++ b/python/cugraph-service/tests/pytest.ini
@@ -0,0 +1,4 @@
+# Copyright (c) 2024, NVIDIA CORPORATION.
+
+[pytest]
+addopts = --tb=native
diff --git a/python/cugraph/pytest.ini b/python/cugraph/pytest.ini
index 675a6cf8fde..bca148538d9 100644
--- a/python/cugraph/pytest.ini
+++ b/python/cugraph/pytest.ini
@@ -17,6 +17,7 @@ addopts =
            --benchmark-max-time=0
            --benchmark-min-rounds=1
            --benchmark-columns="mean, rounds"
+           --tb=native
            ## do not run the slow tests/benchmarks by default
            -m "not slow"
            ## for use with rapids-pytest-benchmark plugin
diff --git a/python/nx-cugraph/nx_cugraph/tests/pytest.ini b/python/nx-cugraph/nx_cugraph/tests/pytest.ini
new file mode 100644
index 00000000000..7b0a9f29fb1
--- /dev/null
+++ b/python/nx-cugraph/nx_cugraph/tests/pytest.ini
@@ -0,0 +1,4 @@
+# Copyright (c) 2024, NVIDIA CORPORATION.
+
+[pytest]
+addopts = --tb=native
diff --git a/python/pylibcugraph/pytest.ini b/python/pylibcugraph/pytest.ini
index 573628de680..d5ade9f4836 100644
--- a/python/pylibcugraph/pytest.ini
+++ b/python/pylibcugraph/pytest.ini
@@ -14,3 +14,5 @@
 [pytest]
 markers =
           cugraph_ops: Tests requiring cugraph-ops
+
+addopts = --tb=native