From a5e0a48f80a3019314bd3edf9e0d5a5057ad9686 Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Mon, 4 Aug 2025 10:04:20 -0500
Subject: [PATCH] Add test test functions for cpp/python exceptions

---
 python/src/py_nvbench.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/python/src/py_nvbench.cpp b/python/src/py_nvbench.cpp
index 30b5dd0..22cfbc6 100644
--- a/python/src/py_nvbench.cpp
+++ b/python/src/py_nvbench.cpp
@@ -725,4 +725,10 @@ PYBIND11_MODULE(_nvbench, m)
     },
     "Run all registered benchmarks",
     py::arg("argv") = py::list());
+
+  m.def("test_cpp_exception", []() { throw nvbench_run_error("Test"); });
+  m.def("test_py_exception", []() {
+    py::set_error(benchmark_exc, "Test");
+    throw py::error_already_set();
+  });
 }