mirror of
https://github.com/andreinechaev/nvcc4jupyter.git
synced 2026-06-13 18:50:47 +05:30
Add tests
This commit is contained in:
Vendored
+60
@@ -0,0 +1,60 @@
|
||||
import glob
|
||||
import os
|
||||
import sys
|
||||
|
||||
sys.path.append(".")
|
||||
|
||||
import pytest
|
||||
from IPython.core.interactiveshell import InteractiveShell
|
||||
|
||||
from v1.v1 import NVCCPlugin
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def shell():
|
||||
return InteractiveShell()
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def plugin(shell: InteractiveShell):
|
||||
return NVCCPlugin(shell=shell)
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def tests_path():
|
||||
return "tests"
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def fixtures_path(tests_path):
|
||||
return os.path.join(tests_path, "fixtures")
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def sample_magic_cu_line():
|
||||
# fmt: off
|
||||
return '--profile --profiler-args "--metrics l1tex__t_sectors_pipe_lsu_mem_global_op_ld.sum"'
|
||||
# fmt: on
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def sample_cuda_fpath(fixtures_path: str):
|
||||
return os.path.join(fixtures_path, "single_file", "hello.cu")
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def sample_cuda_code(sample_cuda_fpath: str):
|
||||
with open(sample_cuda_fpath, "r", encoding="utf-8") as f:
|
||||
return f.read()
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def timeit_regex():
|
||||
return r".+ ± .+ per loop \(mean ± std. dev. of .+ runs, .+ loops each\)"
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def multiple_source_fpaths(fixtures_path: str):
|
||||
pattern_h = os.path.join(fixtures_path, "multiple_files", "*.h")
|
||||
pattern_cu = os.path.join(fixtures_path, "multiple_files", "*.cu")
|
||||
return list(glob.glob(pattern_h)) + list(glob.glob(pattern_cu))
|
||||
Reference in New Issue
Block a user