muMLE/state/test/test_read_dict_keys.py

51 lines
1.2 KiB
Python

import pytest
@pytest.mark.usefixtures("state")
def test_read_dict_keys_no_exists(state):
assert state.read_dict_keys(100000) == None
@pytest.mark.usefixtures("state")
def test_read_dict_keys_simple(state):
a = state.create_node()
b = state.create_node()
c = state.create_nodevalue("f")
d = state.create_edge(a, b)
e = state.create_edge(d, c)
assert a != None
assert b != None
assert c != None
assert d != None
assert e != None
l = state.read_dict_keys(a)
assert l != None
assert set(l) == set([c])
@pytest.mark.usefixtures("state")
def test_read_dict_keys_multi(state):
a = state.create_node()
b = state.create_node()
c = state.create_nodevalue("f")
d = state.create_edge(a, b)
e = state.create_edge(d, c)
assert a != None
assert b != None
assert c != None
assert d != None
assert e != None
g = state.create_node()
h = state.create_nodevalue("k")
i = state.create_edge(a, g)
j = state.create_edge(i, h)
assert g != None
assert h != None
assert i != None
assert j != None
l = state.read_dict_keys(a)
assert l != None
assert set(l) == set([c, h])