muMLE/state/test/fixtures/state.py
Andrei Bondarenko 1931020bb1 Revert "Reorganization"
This reverts commit f44c888a
2021-07-24 21:50:12 +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)