import pytest
from services.bottom.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)