Measure time spent by each histogramm test

Author: antonwolfy

dpnp/tests/test_histogram.py

@@ -1,3 +1,5 @@
+import time
+
 import dpctl
 import numpy
 import pytest
@@ -31,10 +33,14 @@
 @pytest.fixture(autouse=True)
 def setup_each():
     print("\n[Setup] Run before each test")
+    start_time = time.time()
     free_mem = dpctl.utils.intel_device_info(_dev).get("free_memory", None)
     if free_mem:
         print(f"Global memory available: {free_mem}")
     yield
+    end_time = time.time()
+    duration = end_time - start_time
+    print(f"\n[Test Duration] {duration:.4f} seconds")
     print("[Teardown] Run after each test")
 
 

dpnp/tests/third_party/cupy/statistics_tests/test_histogram.py

@@ -1,4 +1,5 @@
 import sys
+import time
 import unittest
 
 import dpctl
@@ -52,10 +53,14 @@ def for_all_dtypes_combination_bincount(names):
 @pytest.fixture(autouse=True)
 def setup_each():
     print("\n[Setup] Run before each test")
+    start_time = time.time()
     free_mem = dpctl.utils.intel_device_info(_dev).get("free_memory", None)
     if free_mem:
         print(f"Global memory available: {free_mem}")
     yield
+    end_time = time.time()
+    duration = end_time - start_time
+    print(f"\n[Test Duration] {duration:.4f} seconds")
     print("[Teardown] Run after each test")