muMLE/state/test/fixtures/state.py
2021-07-13 04:56:08 +02:00

19 lines
475 B
Python

import pytest
from state.pystate import PyState
# from state.rdfstate import RDFState
# from state.neo4jstate import Neo4jState
@pytest.fixture(params=[
(PyState,),
# (RDFState, "http://example.org/#"),
# (Neo4jState,)
])
def state(request):
if len(request.param) > 1:
state = request.param[0](*request.param[1:])
else:
state = request.param[0]()
yield state
# if isinstance(state, Neo4jState):
# state.close(clear=True)