blob: 9a2c9ad2a170a93586edc1a2b095a12c762c5a75 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
import os
import sys
from distutils.core import Command
class RunTests(Command):
"""
A custom command to run tests.
"""
description = "Run the tests"
commands = {
"all": "pytest",
"unit": "pytest tests/unit",
"integration": "pytest tests/integration",
"performance": "pytest tests/performance",
}
user_options = [
("type=", None,
f"""Specify the type of tests to run.
Valid types are {tuple(commands.keys())}.
Default is `all`.""")]
def __init__(self, dist):
"""Initialise the command."""
super().__init__(dist)
def initialize_options(self):
"""Initialise the default values of all the options."""
self.type = "all"
def finalize_options(self):
"""Set final value of all the options once they are processed."""
if self.type not in RunTests.commands.keys():
raise Exception(f"""
Invalid test type (self.type) requested!
Valid types are
{tuple(RunTests.commands.keys())}""")
def run(self):
"""Run the chosen tests"""
print(f"Running {self.type} tests")
os.system(RunTests.commands[self.type])
|
