muMLE/state/test/fixtures/state.py
2021-08-11 14:55:13 +02:00

19 lines
463 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)