import pytest

import npstructures

import tests.npbackend as npb

def pytest_addoption(parser):

parser.addoption("--cupy", action="store_true", default=False,

help="Run cupy tests.")

def pytest_configure(config):

config.addinivalue_line("markers", "cupy: mark test as supported for cupy backend")

if "cupy" in config.getoption("-m"):

import cupy as cp

npstructures.set_backend(cp)

npb.np = cp

npb.np.testing.assert_equal = cp.testing.assert_array_equal