mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-21 06:40:23 +08:00
45 lines
1002 B
Python
45 lines
1002 B
Python
import os
|
|
import random
|
|
import pytest
|
|
|
|
os.environ["PYTHONHASHSEED"] = "0"
|
|
|
|
random.seed(0)
|
|
|
|
try:
|
|
import torch
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
torch.manual_seed(0)
|
|
|
|
try:
|
|
import numpy as np
|
|
except ImportError:
|
|
pass
|
|
else:
|
|
np.random.seed(0)
|
|
|
|
|
|
def pytest_terminal_summary(terminalreporter, exitstatus, config):
|
|
"""Ensure that at least one test is collected. Error out if all tests are skipped."""
|
|
known_types = {
|
|
"failed",
|
|
"passed",
|
|
"skipped",
|
|
"deselected",
|
|
"xfailed",
|
|
"xpassed",
|
|
"warnings",
|
|
"error",
|
|
}
|
|
if (sum(
|
|
len(terminalreporter.stats.get(k, []))
|
|
for k in known_types.difference({"skipped", "deselected"})) == 0):
|
|
terminalreporter.write_sep(
|
|
"!",
|
|
(f"Error: No tests were collected. "
|
|
f"{dict(sorted((k, len(v)) for k, v in terminalreporter.stats.items()))}"),
|
|
)
|
|
pytest.exit("No tests were collected.", returncode=5)
|