Import regression-test context manager on Omni main

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>

tests/helpers/tests/test_assertions.py

@@ -3,6 +3,7 @@
 
 import io
 import wave
+from contextlib import nullcontext
 from types import SimpleNamespace
 
 import numpy as np