MCPcopy Create free account
hub / github.com/google-deepmind/alphageometry / GraphUtilsTest

Class GraphUtilsTest

graph_utils_test.py:23–141  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

21
22
23class GraphUtilsTest(unittest.TestCase):
24
25 def test_cross(self):
26 self.assertEqual(gu.cross([], [1]), [])
27 self.assertEqual(gu.cross([1], []), [])
28 self.assertEqual(gu.cross([1], [2]), [(1, 2)])
29 self.assertEqual(gu.cross([1], [2, 3]), [(1, 2), (1, 3)])
30
31 e1 = [1, 2, 3]
32 e2 = [4, 5]
33 target = [(1, 4), (1, 5), (2, 4), (2, 5), (3, 4), (3, 5)]
34 self.assertEqual(gu.cross(e1, e2), target)
35
36 def test_comb2(self):
37 self.assertEqual(gu.comb2([]), [])
38 self.assertEqual(gu.comb2([1]), [])
39 self.assertEqual(gu.comb2([1, 2]), [(1, 2)])
40 self.assertEqual(gu.comb2([1, 2, 3]), [(1, 2), (1, 3), (2, 3)])
41
42 def test_comb3(self):
43 self.assertEqual(gu.comb3([]), [])
44 self.assertEqual(gu.comb3([1]), [])
45 self.assertEqual(gu.comb3([1, 2]), [])
46 self.assertEqual(gu.comb3([1, 2, 3]), [(1, 2, 3)])
47 self.assertEqual(
48 gu.comb3([1, 2, 3, 4]), [(1, 2, 3), (1, 2, 4), (1, 3, 4), (2, 3, 4)]
49 )
50
51 def test_comb4(self):
52 self.assertEqual(gu.comb4([]), [])
53 self.assertEqual(gu.comb4([1]), [])
54 self.assertEqual(gu.comb4([1, 2]), [])
55 self.assertEqual(gu.comb4([1, 2, 3]), [])
56 self.assertEqual(gu.comb4([1, 2, 3, 4]), [(1, 2, 3, 4)])
57 self.assertEqual(
58 gu.comb4([1, 2, 3, 4, 5]),
59 [(1, 2, 3, 4), (1, 2, 3, 5), (1, 2, 4, 5), (1, 3, 4, 5), (2, 3, 4, 5)],
60 )
61
62 def test_perm2(self):
63 self.assertEqual(gu.perm2([]), [])
64 self.assertEqual(gu.perm2([1]), [])
65 self.assertEqual(gu.perm2([1, 2]), [(1, 2), (2, 1)])
66 self.assertEqual(
67 gu.perm2([1, 2, 3]), [(1, 2), (2, 1), (1, 3), (3, 1), (2, 3), (3, 2)]
68 )
69
70 def test_perm3(self):
71 self.assertEqual(gu.perm3([]), [])
72 self.assertEqual(gu.perm3([1]), [])
73 self.assertEqual(gu.perm3([1, 2]), [])
74 self.assertEqual(
75 gu.perm3([1, 2, 3]),
76 [(1, 2, 3), (1, 3, 2), (2, 1, 3), (2, 3, 1), (3, 1, 2), (3, 2, 1)],
77 )
78 self.assertEqual(
79 gu.perm3([1, 2, 3, 4]),
80 [

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected