def test_alphabet_graph(self):
g = AlphabetGraph()
self.assertTrue(g.has_node('c'))
self.assertFalse(g.has_node('A'))
self.assertFalse(g.has_node(1))
self.assertCountEqual(g.successors_of('a'), ['b'])
self.assertCountEqual(g.successors_of('A'), [])
self.assertCountEqual(g.successors_of('z'), [])
self.assertCountEqual(g.successors_of(1), [])
self.assertCountEqual(g.predecessors_of('a'), [])
self.assertCountEqual(g.predecessors_of('A'), [])
self.assertCountEqual(g.predecessors_of('z'), ['y'])
self.assertCountEqual(g.predecessors_of(1), [])
self.assertCountEqual(g.hops_from_node('a'), [Hop('a', 'b', 1.0)])
self.assertCountEqual(g.hops_to_node('g'), [Hop('f', 'g', 1.0)])
self.assertEqual(g.hop_weight('a', 'b'), 1.0)
self.assertEqual(g.hop_weight('b', 'a'), 0.0)
self.assertEqual(g.hop_weight(1, 'a'), 0.0)
self.assertTrue(g.find_hop('a', 'b'))
self.assertFalse(g.find_hop('a', 'c'))
self.assertEqual(g.find_hop('a', 'b'), Hop('a', 'b', 1.0))
def find_hop(self, nodes, from_node, to_node):
try:
if (nodes.has_node(from_node) and nodes.has_node(to_node)
and ord(from_node) + 1 == ord(to_node)):
return Hop(from_node, to_node, 1.0)
except TypeError:
return None
def hops_to_node(self, nodes, x):
try:
if nodes.has_node(x) and x > 'a':
pred = chr(ord(x) - 1)
yield Hop(pred, x, 1.0)
except TypeError:
pass
def hops_from_node(self, nodes, x):
try:
if nodes.has_node(x) and x < 'z':
succ = chr(ord(x) + 1)
yield Hop(x, succ, 1.0)
except TypeError:
pass
def test_doubled_graph(self):
g0 = Graph(nodes=EnumNodes(['a', 'b', 'o']),
edges=EnumEdges(Hops.from_pairs(('a', 'b'), ('b', 'o'))))
g = Graph.augment(PrefixedGraph(1, g0), PrefixedGraph(2, g0))
self.assertTrue(g.has_node(PrefixedNode(1, 'a')))
self.assertTrue(g.has_node(PrefixedNode(2, 'a')))
self.assertFalse(g.has_node(PrefixedNode(3, 'a')))
self.assertFalse(g.has_node('a'))
self.assertEqual(
g.find_hop(PrefixedNode(1, 'a'), PrefixedNode(1, 'b')),
Hop(PrefixedNode(1, 'a'), PrefixedNode(1, 'b'), 1.0))
self.assertEqual(
g.find_hop(PrefixedNode(2, 'a'), PrefixedNode(2, 'b')),
Hop(PrefixedNode(2, 'a'), PrefixedNode(2, 'b'), 1.0))
self.assertEqual(
g.find_hop(PrefixedNode(1, 'a'), PrefixedNode(2, 'b')), None)
self.assertEqual(g.find_hop('a', 'b'), None)
# .add_edges() from one PrefixedGraph to the other
g2 = g.add_edges(
EnumEdges(
Hops.from_pairs((PrefixedNode(1, 'a'), PrefixedNode(2, 'a')))))
self.assertEqual(
g2.find_hop(PrefixedNode(1, 'a'), PrefixedNode(2, 'a')),
Hop(PrefixedNode(1, 'a'), PrefixedNode(2, 'a'), 1.0))
self.assertCountEqual(g2.query(OfClass(str)), [
PrefixedNode(1, 'a'),
PrefixedNode(1, 'b'),
PrefixedNode(1, 'o'),
PrefixedNode(2, 'a'),
PrefixedNode(2, 'b'),
PrefixedNode(2, 'o')
])
self.assertCountEqual(
g2.query(WithPrefix(2, OfClass(str))),
[PrefixedNode(2, 'a'),
PrefixedNode(2, 'b'),
PrefixedNode(2, 'o')])
self.assertCountEqual(g0.query(WithPrefix(2, OfClass(str))), [])
def test_features(self):
@dataclass(frozen=True)
class Parity(Feature):
name: str
Even = Parity('Even')
Odd = Parity('Odd')
@dataclass(frozen=True)
class BaseNode:
n: int
def features_of(self):
if self.n & 1:
yield Odd
else:
yield Even
yield self.n
g = Graph.with_features([BaseNode(1), BaseNode(2), BaseNode(3)])
self.assertCountEqual(g.query(OfClass(Parity)), [Odd, Even])
self.assertCountEqual(g.hops_to_node(BaseNode(1)), [
Hop(Odd, BaseNode(1), 1.0),
Hop(1, BaseNode(1), 1.0),
Hop(BaseNode, BaseNode(1), 1.0)
])
self.assertCountEqual(g.hops_from_node(Odd), [
Hop(Odd, BaseNode(1), 1.0),
Hop(Odd, BaseNode(3), 1.0),
Hop(Odd, Parity, 1.0)
])
self.assertEqual(g.find_hop(BaseNode(2), Even),
Hop(BaseNode(2), Even, 1.0))
self.assertEqual(g.find_hop(Even, Odd), None)
# Now add mutual inhibition
g = g.add_edges(MutualInhibition((Feature, int)))
self.assertEqual(g.find_hop(Even, Odd), Hop(Even, Odd, -0.2))
self.assertEqual(g.find_hop(Odd, Even), Hop(Odd, Even, -0.2))
self.assertEqual(g.find_hop(1, 2), Hop(1, 2, -0.2))
def test_prefix(self):
g0 = Graph(nodes=EnumNodes(['a', 'b', 'o']),
edges=EnumEdges(Hops.from_pairs(('a', 'b'), ('b', 'o'))))
g = PrefixedGraph(2, g0)
# methods that go to Nodes
self.assertFalse(g.has_node('a'))
self.assertTrue(g.has_node(PrefixedNode(2, 'a')))
self.assertFalse(g.has_node(PrefixedNode(1, 'a')))
self.assertCountEqual(
g.query(OfClass(str)),
[PrefixedNode(2, 'a'),
PrefixedNode(2, 'b'),
PrefixedNode(2, 'o')])
# methods that go to Edges
# .hops_from_node()
self.assertCountEqual(g.hops_from_node('a'), [])
self.assertCountEqual(g.hops_from_node(PrefixedNode(
2, 'a')), [Hop(PrefixedNode(2, 'a'), PrefixedNode(2, 'b'), 1.0)])
self.assertCountEqual(g.hops_from_node(PrefixedNode(1, 'a')), [])
# .hops_to_node()
self.assertCountEqual(g.hops_to_node('b'), [])
self.assertCountEqual(g.hops_to_node(PrefixedNode(
2, 'b')), [Hop(PrefixedNode(2, 'a'), PrefixedNode(2, 'b'), 1.0)])
self.assertCountEqual(g.hops_to_node(PrefixedNode(1, 'b')), [])
# .find_hop()
self.assertIsNone(g.find_hop('a', 'b'))
self.assertEqual(
g.find_hop(PrefixedNode(2, 'a'), PrefixedNode(2, 'b')),
Hop(PrefixedNode(2, 'a'), PrefixedNode(2, 'b'), 1.0))
self.assertEqual(
g.find_hop(PrefixedNode(1, 'a'), PrefixedNode(2, 'b')), None)
self.assertEqual(
g.find_hop(PrefixedNode(2, 'a'), PrefixedNode(1, 'b')), None)
# .degree_out() and .degree_in()
self.assertEqual(g.degree_out('a'), 0)
self.assertEqual(g.degree_out(PrefixedNode(1, 'a')), 0)
self.assertEqual(g.degree_out(PrefixedNode(2, 'a')), 1)
self.assertEqual(g.degree_in('b'), 0)
self.assertEqual(g.degree_in(PrefixedNode(1, 'b')), 0)
self.assertEqual(g.degree_in(PrefixedNode(2, 'b')), 1)
# .successors_of() and .predecessors_of()
self.assertCountEqual(g.successors_of('a'), [])
self.assertCountEqual(g.successors_of(PrefixedNode(1, 'a')), [])
self.assertCountEqual(g.successors_of(PrefixedNode(2, 'a')),
[PrefixedNode(2, 'b')])
self.assertCountEqual(g.predecessors_of(PrefixedNode(2, 'b')),
[PrefixedNode(2, 'a')])
# .hop_weight()
self.assertEqual(
g.hop_weight(PrefixedNode(2, 'a'), PrefixedNode(2, 'b')), 1.0)
self.assertEqual(
g.hop_weight(PrefixedNode(1, 'a'), PrefixedNode(2, 'b')), 0.0)
self.assertEqual(
g.hop_weight(PrefixedNode(2, 'a'), PrefixedNode(1, 'b')), 0.0)
# .add_edges()
g2 = g.add_edges(
EnumEdges(
Hops.from_pairs((PrefixedNode(2, 'a'), PrefixedNode(2, 'o')))))
self.assertEqual(
g2.find_hop(PrefixedNode(2, 'a'), PrefixedNode(2, 'o')),
Hop(PrefixedNode(2, 'a'), PrefixedNode(2, 'o'), 1.0))
self.assertEqual(g2.find_hop('a', 'o'), None)