def test_find(self):
identi = lambda x: x
idents = lambda x: x
spec = GraphSpec()
spec.add_identifier(int, identi)
spec.add_identifier(str, idents)
assert spec.get_identifier(int) is identi
assert spec.get_identifier(str) is idents
assert spec.get_identifier(float) is None
def test_inherit(self):
class A:
pass
class B(A):
pass
ident = lambda x: x
spec = GraphSpec()
spec.add_identifier(A, ident)
assert spec.get_identifier(B) is ident