HugoVoxx commited on
Commit
3e02422
·
verified ·
1 Parent(s): eb6f08a

Delete ag4masses/alphageometry/graph_utils_test.py

Browse files
ag4masses/alphageometry/graph_utils_test.py DELETED
@@ -1,145 +0,0 @@
1
- # Copyright 2023 DeepMind Technologies Limited
2
- #
3
- # Licensed under the Apache License, Version 2.0 (the "License");
4
- # you may not use this file except in compliance with the License.
5
- # You may obtain a copy of the License at
6
- #
7
- # http://www.apache.org/licenses/LICENSE-2.0
8
- #
9
- # Unless required by applicable law or agreed to in writing, software
10
- # distributed under the License is distributed on an "AS IS" BASIS,
11
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
- # See the License for the specific language governing permissions and
13
- # limitations under the License.
14
- # ==============================================================================
15
-
16
- """Unit tests for graph_utils.py."""
17
- import unittest
18
-
19
- from absl.testing import absltest
20
- import graph_utils as gu
21
-
22
-
23
- class 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
- [
81
- (1, 2, 3),
82
- (1, 2, 4),
83
- (1, 3, 2),
84
- (1, 3, 4),
85
- (1, 4, 2),
86
- (1, 4, 3),
87
- (2, 1, 3),
88
- (2, 1, 4),
89
- (2, 3, 1),
90
- (2, 3, 4),
91
- (2, 4, 1),
92
- (2, 4, 3),
93
- (3, 1, 2),
94
- (3, 1, 4),
95
- (3, 2, 1),
96
- (3, 2, 4),
97
- (3, 4, 1),
98
- (3, 4, 2),
99
- (4, 1, 2),
100
- (4, 1, 3),
101
- (4, 2, 1),
102
- (4, 2, 3),
103
- (4, 3, 1),
104
- (4, 3, 2),
105
- ],
106
- )
107
-
108
- def test_perm4(self):
109
- self.assertEqual(gu.perm3([]), [])
110
- self.assertEqual(gu.perm3([1]), [])
111
- self.assertEqual(gu.perm3([1, 2]), [])
112
- self.assertEqual(gu.perm4([1, 2, 3]), [])
113
- self.assertEqual(
114
- gu.perm4([1, 2, 3, 4]),
115
- [
116
- (1, 2, 3, 4),
117
- (1, 2, 4, 3),
118
- (1, 3, 2, 4),
119
- (1, 3, 4, 2),
120
- (1, 4, 2, 3),
121
- (1, 4, 3, 2), # pylint: disable=line-too-long
122
- (2, 1, 3, 4),
123
- (2, 1, 4, 3),
124
- (2, 3, 1, 4),
125
- (2, 3, 4, 1),
126
- (2, 4, 1, 3),
127
- (2, 4, 3, 1), # pylint: disable=line-too-long
128
- (3, 1, 2, 4),
129
- (3, 1, 4, 2),
130
- (3, 2, 1, 4),
131
- (3, 2, 4, 1),
132
- (3, 4, 1, 2),
133
- (3, 4, 2, 1), # pylint: disable=line-too-long
134
- (4, 1, 2, 3),
135
- (4, 1, 3, 2),
136
- (4, 2, 1, 3),
137
- (4, 2, 3, 1),
138
- (4, 3, 1, 2),
139
- (4, 3, 2, 1),
140
- ], # pylint: disable=line-too-long
141
- )
142
-
143
-
144
- if __name__ == '__main__':
145
- absltest.main()