Add test expression option to filter tests by their names

Creates an interface for pytest test selection mechanism.

Signed-off-by: Sławomir Błauciak <slawomir.blauciak@intel.com>

Author: slawblauciak

scripts/test-triton.sh

@@ -45,6 +45,7 @@ OPTION:
     --skip-list SKIPLIST
     --extra-skip-list-suffixes SEMICOLON-SEPARATED LIST OF SUFFIXES
     --select-from-file SELECTFILE
+    --test-expr EXPRESSION
 "
 
 err() {
@@ -86,6 +87,7 @@ TRITON_TEST_IGNORE_ERRORS=false
 SKIP_PIP=false
 SKIP_PYTORCH=false
 TEST_UNSKIP=false
+TRITON_TEST_EXPRESSION=""
 
 while (( $# != 0 )); do
   case "$1" in
@@ -273,6 +275,10 @@ while (( $# != 0 )); do
       TRITON_TEST_SELECTFILE="$(realpath "$2")"
       shift 2
       ;;
+    --test-expr)
+      TRITON_TEST_EXPRESSION="$2"
+      shift 2
+      ;;
     --help)
       echo "$HELP"
       exit 0
@@ -342,12 +348,18 @@ run_unit_tests() {
 }
 
 run_pytest_command() {
+  local pytest_args=("$@")
+
+  if [[ -n "$TRITON_TEST_EXPRESSION" ]]; then
+    pytest_args+=(-k "$TRITON_TEST_EXPRESSION")
+  fi
+
   if [[ -n "$TRITON_TEST_SELECTFILE" ]]; then
-    if pytest "$@" --collect-only > /dev/null 2>&1; then
-      pytest "$@"
+    if pytest "${pytest_args[@]}" --collect-only > /dev/null 2>&1; then
+      pytest "${pytest_args[@]}"
     fi
   else
-    pytest "$@"
+    pytest "${pytest_args[@]}"
   fi
 }