-#!/bin/bash
-
-DOWNLOAD_TESTS="age_restriction|download|subtitles|write_annotations|iqiyi_sdk_interpreter"
-
-test_set=""
-
-case "$YTDL_TEST_SET" in
- core)
- test_set="-I test_($DOWNLOAD_TESTS)\.py"
- ;;
- download)
- test_set="-I test_(?!$DOWNLOAD_TESTS).+\.py"
- ;;
- *)
- break
- ;;
-esac
-
-nosetests test --verbose $test_set
+#!/bin/sh
+
+if [ -z $1 ]; then
+ test_set='test'
+elif [ $1 = 'core' ]; then
+ test_set="-m not download"
+elif [ $1 = 'download' ]; then
+ test_set="-m download"
+else
+ echo 'Invalid test type "'$1'". Use "core" | "download"'
+ exit 1
+fi
+
+python3 -bb -Werror -m pytest "$test_set"