Domovská stránka Prehľadávať Pomoc
Registrovať Prihlásiť sa
olinox14
/
pypog
1
0
Fork 0
Súbory Issues 0 Pull requesty 0 Wiki
Strom: 17dae6af8a
Branche Tagy
pypog
/
tests
/
test_geometry.py

test_geometry.py 19 KB
História Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335 
  1. '''
    2. 

  2. 

  3.     ** By Cro-Ki l@b, 2017 **
    4. 

  4. '''
    5. 

  5. import unittest
    6. 

  6. 

  7. from pypog import geometry
    8. 

  8. 

  9. 

  10. class Test(unittest.TestCase):
    11. 

  11. 

  12.     def test_line_errors(self):
    13. 

  13.         self.assertRaises(TypeError, geometry.line2d, geometry.FLAT_HEX, "a", 1, 1, 1)
    14. 

  14.         self.assertRaises(TypeError, geometry.line2d, geometry.FLAT_HEX, 1, "a", 1, 1)
    15. 

  15.         self.assertRaises(TypeError, geometry.line2d, geometry.FLAT_HEX, 1, 1, "a", 1)
    16. 

  16.         self.assertRaises(TypeError, geometry.line2d, geometry.FLAT_HEX, 1, 1, 1, "a")
    17. 

  17.         self.assertRaises(ValueError, geometry.line2d, 0, 1, 1, 1, 1)
    18. 

  18. 

  19.         self.assertRaises(TypeError, geometry.line3d, geometry.FLAT_HEX, 1, 1, "a", 1, 1, 1)
    20. 

  20.         self.assertRaises(TypeError, geometry.line3d, geometry.FLAT_HEX, 1, 1, 1, 1, 1, "a")
    21. 

  21. 

  22.     def test_line2d(self):
    23. 

  23.         """ 2d line on square or hexagonal grid """
    24. 

  24.         cell_shape = geometry.FLAT_HEX
    25. 

  25. 

  26.         attended = {
    27. 

  27.                     geometry.FLAT_HEX:    {
    28. 

  28.                                       (1, 1, 1, 1): [(1, 1)],
    29. 

  29.                                       (0, 0, 1, 1): [(0, 0), (0, 1), (1, 1)],
    30. 

  30.                                       (1, 1, 0, 0): [(1, 1), (0, 1), (0, 0)],
    31. 

  31.                                       (0, 0, 7, 3): [(0, 0), (1, 0), (2, 1), (3, 1), (4, 2), (5, 2), (6, 3), (7, 3)],
    32. 

  32.                                       (7, 3, 0, 0): [(7, 3), (6, 3), (5, 2), (4, 2), (3, 1), (2, 1), (1, 0), (0, 0)],
    33. 

  33.                                       (4, 3, 0, 3): [(4, 3), (3, 2), (2, 3), (1, 2), (0, 3)],
    34. 

  34.                                       (0, 3, 4, 3): [(0, 3), (1, 2), (2, 3), (3, 2), (4, 3)],
    35. 

  35.                                       (3, 0, 3, 3): [(3, 0), (3, 1), (3, 2), (3, 3)],
    36. 

  36.                                       (3, 3, 3, 0): [(3, 3), (3, 2), (3, 1), (3, 0)]
    37. 

  37.                                      },
    38. 

  38. 

  39.                     geometry.SQUARE: {
    40. 

  40.                                       (1, 1, 1, 1): [(1, 1)],
    41. 

  41.                                       (0, 0, 0, 1): [(0, 0), (0, 1)],
    42. 

  42.                                       (0, 1, 0, 0): [(0, 1), (0, 0)],
    43. 

  43.                                       (0, 0, 1, 1): [(0, 0), (1, 1)],
    44. 

  44.                                       (1, 1, 0, 0): [(1, 1), (0, 0)],
    45. 

  45.                                       (0, 0, 7, 3): [(0, 0), (1, 0), (2, 1), (3, 1), (4, 2), (5, 2), (6, 3), (7, 3)],
    46. 

  46.                                       (7, 3, 0, 0): [(7, 3), (6, 3), (5, 2), (4, 2), (3, 1), (2, 1), (1, 0), (0, 0)],
    47. 

  47.                                       (4, 3, 0, 3): [(4, 3), (3, 3), (2, 3), (1, 3), (0, 3)],
    48. 

  48.                                       (0, 3, 4, 3): [(0, 3), (1, 3), (2, 3), (3, 3), (4, 3)],
    49. 

  49.                                       (3, 0, 3, 3): [(3, 0), (3, 1), (3, 2), (3, 3)],
    50. 

  50.                                       (3, 3, 3, 0): [(3, 3), (3, 2), (3, 1), (3, 0)]
    51. 

  51.                                      }
    52. 

  52.                    }
    53. 

  53. 

  54.         for cell_shape, tests in attended.items():
    55. 

  55.             for args, result in tests.items():
    56. 

  56.                 line = geometry.line2d(cell_shape, *args)
    57. 

  57.                 self.assertEqual(line, result)
    58. 

  58. 

  59.     def test_line3d(self):
    60. 

  60.         """ 3d line on hexagonal and square grid """
    61. 

  61.         cell_shape = geometry.FLAT_HEX
    62. 

  62. 

  63.         attended = {
    64. 

  64.                     geometry.FLAT_HEX:    {
    65. 

  65.                                       (1, 1, 1, 1, 1, 1) : [(1, 1, 1)],
    66. 

  66.                                       (1, 1, 0, 1, 1, 1) : [(1, 1, 0), (1, 1, 1)],
    67. 

  67.                                       (0, 0, 0, 1, 1, 1) : [(0, 0, 0), (0, 1, 0), (1, 1, 1)],
    68. 

  68.                                       (0, 0, 0, 7, 3, 7) : [(0, 0, 0), (1, 0, 1), (2, 1, 2), (3, 1, 3), (4, 2, 4), (5, 2, 5), (6, 3, 6), (7, 3, 7)],
    69. 

  69.                                       (4, 3, 10, 0, 3, 3): [(4, 3, 10), (3, 2, 9), (3, 2, 8), (2, 3, 7), (2, 3, 6), (1, 2, 5), (1, 2, 4), (0, 3, 3)],
    70. 

  70.                                       (3, 0, 0, 3, 3, 0) : [(3, 0, 0), (3, 1, 0), (3, 2, 0), (3, 3, 0)]
    71. 

  71.                                      },
    72. 

  72. 

  73.                     geometry.SQUARE: {
    74. 

  74.                                       (1, 1, 1, 1, 1, 1) : [(1, 1, 1)],
    75. 

  75.                                       (1, 1, 0, 1, 1, 1) : [(1, 1, 0), (1, 1, 1)],
    76. 

  76.                                       (0, 0, 0, 1, 1, 1) : [(0, 0, 0), (1, 1, 1)],
    77. 

  77.                                       (0, 0, 0, 7, 3, 7) : [(0, 0, 0), (1, 0, 1), (2, 1, 2), (3, 1, 3), (4, 2, 4), (5, 2, 5), (6, 3, 6), (7, 3, 7)],
    78. 

  78.                                       (4, 3, 10, 0, 3, 3): [(4, 3, 10), (3, 3, 9), (3, 3, 8), (2, 3, 7), (2, 3, 6), (1, 3, 5), (1, 3, 4), (0, 3, 3)],
    79. 

  79.                                       (3, 0, 0, 3, 3, 0) : [(3, 0, 0), (3, 1, 0), (3, 2, 0), (3, 3, 0)]
    80. 

  80.                                      }
    81. 

  81.                    }
    82. 

  82. 

  83.         for cell_shape, tests in attended.items():
    84. 

  84.             for args, result in tests.items():
    85. 

  85.                 line = geometry.line3d(cell_shape, *args)
    86. 

  86.                 self.assertEqual(line, result)
    87. 

  87. 

  88.     def test_neighbours(self):
    89. 

  89.         for coord in ((0, 0), (-10, -10), (10, 10)):
    90. 

  90.             x, y = coord
    91. 

  91.             self.assertEqual(geometry.neighbours(geometry.FLAT_HEX, x, y), geometry.fhex_neighbours(x, y))
    92. 

  92.             self.assertEqual(geometry.neighbours(geometry.SQUARE, x, y), geometry.squ_neighbours(x, y))
    93. 

  93. 

  94.     def test_fhex_neighbours(self):
    95. 

  95.         self.assertCountEqual(geometry.fhex_neighbours(3, 3), [(3, 2), (4, 3), (4, 4), (3, 4), (2, 4), (2, 3)])
    96. 

  96.         self.assertCountEqual(geometry.fhex_neighbours(4, 4), [(4, 3), (5, 3), (5, 4), (4, 5), (3, 4), (3, 3)])
    97. 

  97. 

  98.     def test_squ_neighbours(self):
    99. 

  99.         self.assertCountEqual(geometry.squ_neighbours(3, 3), [(2, 3), (2, 2), (3, 2), (4, 2), (4, 3), (4, 4), (3, 4), (2, 4)])
    100. 

  100. 

  101.     def test_pivot_errors(self):
    102. 

  102. 

  103.         # invalid cell shape
    104. 

  104.         self.assertRaises(ValueError, geometry.pivot, 0, (0, 0), [(0, 0)], 1)
    105. 

  105. 

  106.         self.assertRaises(TypeError, geometry.pivot, 0, "a"    , [(0, 0)], 1)
    107. 

  107.         self.assertRaises(ValueError, geometry.pivot, 0, ("a", 0), [(0, 0)], 1)
    108. 

  108. 

  109.         self.assertRaises(TypeError, geometry.pivot, 0, (0, 0), 0, 1)
    110. 

  110.         self.assertRaises(ValueError, geometry.pivot, 0, (0, 0), ["a", (0, 0)], 1)
    111. 

  111.         self.assertRaises(ValueError, geometry.pivot, 0, (0, 0), [("a", 0), (0, 0)], 1)
    112. 

  112. 

  113.         self.assertRaises(TypeError, geometry.pivot, 0, (0, 0), 1, "a")
    114. 

  114. 

  115.     def test_hex_pivot(self):
    116. 

  116.         """ pivot on hexagonal grid """
    117. 

  117. 

  118.         attended = [
    119. 

  119.                     [(5, 5), (4, 5), (6, 6)],
    120. 

  120.                     [(5, 6), (4, 7), (6, 6)],
    121. 

  121.                     [(6, 7), (6, 8), (6, 6)],
    122. 

  122.                     [(7, 6), (8, 7), (6, 6)],
    123. 

  123.                     [(7, 5), (8, 5), (6, 6)],
    124. 

  124.                     [(6, 5), (6, 4), (6, 6)],
    125. 

  125.                     [(5, 5), (4, 5), (6, 6)]
    126. 

  126.                    ]
    127. 

  127. 

  128. 

  129.         for i in range(len(attended)):
    130. 

  130.             self.assertCountEqual(geometry.pivot(geometry.FLAT_HEX, (6, 6), [(6, 6)], i), [(6, 6)])
    131. 

  131.             result = geometry.pivot(geometry.FLAT_HEX, (6, 6), [(5, 5), (4, 5), (6, 6)], i)
    132. 

  132.             self.assertCountEqual(result, attended[i])
    133. 

  133. 

  134.     def test_squ_pivot(self):
    135. 

  135.         """ pivot on square grid """
    136. 

  136.         attended = [
    137. 

  137.                     [(6, 6), (6, 5), (5, 5), (5, 6)],
    138. 

  138.                     [(6, 6), (5, 6), (5, 7), (6, 7)],
    139. 

  139.                     [(6, 6), (6, 7), (7, 7), (7, 6)],
    140. 

  140.                     [(6, 6), (7, 6), (7, 5), (6, 5)],
    141. 

  141.                     [(6, 6), (6, 5), (5, 5), (5, 6)]
    142. 

  142.                    ]
    143. 

  143. 

  144.         for i in range(len(attended)):
    145. 

  145.             self.assertCountEqual(geometry.pivot(geometry.SQUARE, (6, 6), [(6, 6)], i), [(6, 6)])
    146. 

  146.             result = geometry.pivot(geometry.SQUARE, (6, 6), [(6, 6), (6, 5), (5, 5), (5, 6)], i)
    147. 

  147.             self.assertCountEqual(result, attended[i])
    148. 

  148. 

  149. 

  150.     def test_rectangle_errors(self):
    151. 

  151.         for method in (geometry.rectangle, geometry.hollow_rectangle):
    152. 

  152.             self.assertRaises(TypeError, method, "a", 1, 1, 1)
    153. 

  153.             self.assertRaises(TypeError, method, 1, "a", 1, 1)
    154. 

  154.             self.assertRaises(TypeError, method, 1, 1, "a", 1)
    155. 

  155.             self.assertRaises(TypeError, method, 1, 1, 1, "a")
    156. 

  156. 

  157.     def test_rectangle(self):
    158. 

  158. 

  159.         self.assertEquals(geometry.rectangle(0, 0, 0, 0), [(0, 0)])
    160. 

  160.         self.assertCountEqual(geometry.rectangle(0, 0, 1, 1), [(0, 0), (0, 1), (1, 1), (1, 0)])
    161. 

  161.         self.assertCountEqual(geometry.rectangle(1, 1, 0, 0), [(0, 0), (0, 1), (1, 1), (1, 0)])
    162. 

  162.         self.assertCountEqual(geometry.rectangle(4, 3, 7, 5), [(4, 3), (4, 4), (4, 5), (5, 5), (6, 5), (7, 5), (7, 4), (7, 3), (6, 3), (5, 3), (6, 4), (5, 4)])
    163. 

  163.         self.assertCountEqual(geometry.rectangle(3, 3, 9, 9), [(3, 3), (9, 9), (9, 8), (9, 7), (9, 5), (9, 6), (9, 4), (9, 3), (8, 4), (7, 3), (6, 4), (4, 4),
    164. 

  164.                                                        (3, 4), (3, 5), (3, 6), (3, 7), (3, 8), (3, 9), (4, 9), (5, 9), (6, 9), (7, 9), (8, 9), (4, 5),
    165. 

  165.                                                        (5, 4), (6, 5), (7, 4), (8, 5), (4, 6), (5, 5), (6, 6), (7, 5), (8, 6), (4, 7), (5, 6), (6, 7),
    166. 

  166.                                                        (7, 6), (8, 7), (4, 8), (5, 7), (6, 8), (7, 7), (8, 8), (7, 8), (5, 8), (8, 3), (6, 3), (4, 3),
    167. 

  167.                                                        (5, 3)])
    168. 

  168. 

  169.         self.assertEquals(geometry.hollow_rectangle(0, 0, 0, 0), [(0, 0)])
    170. 

  170.         self.assertCountEqual(geometry.hollow_rectangle(0, 0, 1, 1), [(0, 0), (0, 1), (1, 1), (1, 0)])
    171. 

  171.         self.assertCountEqual(geometry.hollow_rectangle(1, 1, 0, 0), [(0, 0), (0, 1), (1, 1), (1, 0)])
    172. 

  172.         self.assertCountEqual(geometry.hollow_rectangle(4, 3, 7, 5), [(4, 3), (4, 4), (4, 5), (5, 5), (6, 5), (7, 5), (7, 4), (7, 3), (6, 3), (5, 3)])
    173. 

  173.         self.assertCountEqual(geometry.hollow_rectangle(3, 3, 9, 9), [(3, 3), (9, 9), (9, 8), (9, 7), (9, 5), (9, 6), (9, 4), (9, 3), (7, 3), (3, 4),
    174. 

  174.                                                               (3, 5), (3, 6), (3, 7), (3, 8), (3, 9), (4, 9), (5, 9), (6, 9), (7, 9), (8, 9),
    175. 

  175.                                                               (8, 3), (6, 3), (4, 3), (5, 3)])
    176. 

  176. 

  177.     def test_triangle_errors(self):
    178. 

  178. 

  179.         for cell_shape in (geometry.FLAT_HEX, geometry.SQUARE):
    180. 

  180.             self.assertRaises(TypeError, geometry.triangle, cell_shape, "a", 1, 1, 1, 1)
    181. 

  181.             self.assertRaises(TypeError, geometry.triangle, cell_shape, 1, "a", 1, 1, 1)
    182. 

  182.             self.assertRaises(TypeError, geometry.triangle, cell_shape, 1, 1, "a", 1, 1)
    183. 

  183.             self.assertRaises(TypeError, geometry.triangle, cell_shape, 1, 1, 1, "a", 1)
    184. 

  184.             self.assertRaises(ValueError, geometry.triangle, cell_shape, 1, 1, 1, 1, -1)
    185. 

  185. 

  186.             self.assertRaises(TypeError, geometry.triangle3d, cell_shape, "a", 1, 1, 1, 1, 1, 1)
    187. 

  187.             self.assertRaises(TypeError, geometry.triangle3d, cell_shape, 1, "a", 1, 1, 1, 1, 1)
    188. 

  188.             self.assertRaises(TypeError, geometry.triangle3d, cell_shape, 1, 1, "a", 1, 1, 1, 1)
    189. 

  189.             self.assertRaises(TypeError, geometry.triangle3d, cell_shape, 1, 1, 1, "a", 1, 1, 1)
    190. 

  190.             self.assertRaises(TypeError, geometry.triangle3d, cell_shape, 1, 1, 1, 1, "a", 1, 1)
    191. 

  191.             self.assertRaises(TypeError, geometry.triangle3d, cell_shape, 1, 1, 1, 1, 1, "a", 1)
    192. 

  192.             self.assertRaises(ValueError, geometry.triangle3d, cell_shape, 1, 1, 1, 1, 1, 1, -1)
    193. 

  193. 

  194.         self.assertRaises(ValueError, geometry.triangle, 0, 1, 1, 1, 1, 1)
    195. 

  195.         self.assertRaises(ValueError, geometry.triangle3d, 0, 1, 1, 1, 1, 1, 1, 1)
    196. 

  196. 

  197.     def test_squ2_triangle(self):
    198. 

  198.         """test triangle algorithms on square grid"""
    199. 

  199.         cell_shape = geometry.SQUARE
    200. 

  200. 

  201.         for i in geometry.ANGLES:
    202. 

  202.             self.assertCountEqual(geometry.triangle(cell_shape, 0, 0, 0, 0, i), [(0, 0)])
    203. 

  203. 

  204.         # TODO: check and validate
    205. 

  205. #         # left to right
    206. 

  206. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    207. 

  207. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    208. 

  208. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    209. 

  209. #
    210. 

  210. #         # top to bottom
    211. 

  211. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    212. 

  212. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    213. 

  213. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    214. 

  214. #
    215. 

  215. #         # right to left
    216. 

  216. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    217. 

  217. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    218. 

  218. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    219. 

  219. #
    220. 

  220. #         # bottom to top
    221. 

  221. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    222. 

  222. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    223. 

  223. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    224. 

  224. #
    225. 

  225. #         # top left to bottom right
    226. 

  226. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    227. 

  227. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    228. 

  228. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    229. 

  229. #
    230. 

  230. #         # bottom right to top left
    231. 

  231. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    232. 

  232. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    233. 

  233. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    234. 

  234. #
    235. 

  235. #         # top right to bottom left
    236. 

  236. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    237. 

  237. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    238. 

  238. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    239. 

  239. #
    240. 

  240. #         # bottom right to top left
    241. 

  241. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    242. 

  242. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    243. 

  243. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    244. 

  244. 

  245. 

  246. 

  247.     def test_fhex2_triangle(self):
    248. 

  248.         """test triangle algorithms on hexagonal grid"""
    249. 

  249.         cell_shape = geometry.FLAT_HEX
    250. 

  250.         for i in geometry.ANGLES:
    251. 

  251.             self.assertCountEqual(geometry.triangle(cell_shape, 0, 0, 0, 0, i), [(0, 0)])
    252. 

  252. 

  253.         # left to right
    254. 

  254.         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [(3, 3), (3, 4), (3, 3), (4, 5), (4, 4), (4, 3), (4, 2), (4, 1), (4, 1), (3, 1), (3, 2), (2, 3)])
    255. 

  255.         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [(3, 3), (4, 4), (4, 3), (4, 2), (4, 2), (3, 2), (2, 3)])
    256. 

  256.         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [(3, 3), (4, 4), (4, 3), (4, 2), (4, 2), (3, 2), (2, 3)])
    257. 

  257. 

  258.         # TODO: check and validate
    259. 

  259. 

  260. #         # top to bottom
    261. 

  261. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    262. 

  262. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    263. 

  263. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    264. 

  264. #
    265. 

  265.         # right to left
    266. 

  266.         self.assertCountEqual(geometry.triangle(cell_shape, 4, 3, 2, 3, 1), [(3, 2), (3, 1), (3, 2), (2, 1), (2, 2), (2, 3), (2, 4), (2, 5), (2, 5), (3, 4), (3, 3), (4, 3)])
    267. 

  267.         self.assertCountEqual(geometry.triangle(cell_shape, 4, 3, 2, 3, 2), [(3, 2), (2, 2), (2, 3), (2, 4), (2, 4), (3, 3), (4, 3)])
    268. 

  268.         self.assertCountEqual(geometry.triangle(cell_shape, 4, 3, 2, 3, 3), [(3, 2), (2, 2), (2, 3), (2, 4), (2, 4), (3, 3), (4, 3)])
    269. 

  269. 

  270. #         # bottom to top
    271. 

  271. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    272. 

  272. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    273. 

  273. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    274. 

  274. #
    275. 

  275. #         # top left to bottom right
    276. 

  276. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    277. 

  277. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    278. 

  278. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    279. 

  279. #
    280. 

  280. #         # bottom right to top left
    281. 

  281. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    282. 

  282. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    283. 

  283. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    284. 

  284. #
    285. 

  285. #         # top right to bottom left
    286. 

  286. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    287. 

  287. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    288. 

  288. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    289. 

  289. #
    290. 

  290. #         # bottom right to top left
    291. 

  291. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 1), [])
    292. 

  292. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 2), [])
    293. 

  293. #         self.assertCountEqual(geometry.triangle(cell_shape, 2, 3, 4, 3, 3), [])
    294. 

  294. 

  295. 

  296. 

  297.     def test_squ3_triangle(self):
    298. 

  298.         """test triangle3d algorithms on square grid"""
    299. 

  299.         cell_shape = geometry.SQUARE
    300. 

  300.         # TODO: complete
    301. 

  301. 

  302.     def test_fhex3_triangle(self):
    303. 

  303.         """test triangle3d algorithms on hexagonal grid"""
    304. 

  304.         cell_shape = geometry.FLAT_HEX
    305. 

  305.         # TODO: complete
    306. 

  306. 

  307.     def test_zone(self):
    308. 

  308.         """test the errors due to bad parameters"""
    309. 

  309.         self.assertRaises(TypeError, geometry.zone, 5, 0, 0, "a")
    310. 

  310.         self.assertRaises(TypeError, geometry.zone, 5, "a", 0, 1)
    311. 

  311.         self.assertRaises(TypeError, geometry.zone, 5, 0, "a", 1)
    312. 

  312.         self.assertRaises(ValueError, geometry.zone, 5, 0, 0, -1)
    313. 

  313.         self.assertRaises(ValueError, geometry.zone, -1, 0, 0, 1)
    314. 

  314.         self.assertRaises(ValueError, geometry.zone, "a", 0, 0, 1)
    315. 

  315. 

  316.     def test_hex_zone(self):
    317. 

  317.         """ test the zone algo for hexagonal grid """
    318. 

  318.         cell_shape = geometry.FLAT_HEX
    319. 

  319.         self.assertCountEqual(geometry.zone(cell_shape, 3, 3, 0), [(3, 3)])
    320. 

  320.         self.assertCountEqual(geometry.zone(cell_shape, 3, 3, 1), [(3, 2), (2, 3), (3, 3), (4, 3), (4, 4), (3, 4), (2, 4)])
    321. 

  321.         self.assertCountEqual(geometry.zone(cell_shape, 3, 3, 2), [(3, 2), (1, 3), (5, 4), (4, 5), (1, 4), (2, 3), (4, 2), \
    322. 

  322.                                                             (2, 5), (5, 3), (1, 2), (3, 5), (3, 3), (4, 4), (3, 1), \
    323. 

  323.                                                             (4, 3), (2, 2), (3, 4), (2, 4), (5, 2)])
    324. 

  324. 

  325.     def test_squ_zone(self):
    326. 

  326.         """ test the zone algo for square grid """
    327. 

  327.         cell_shape = geometry.SQUARE
    328. 

  328.         self.assertCountEqual(geometry.zone(cell_shape, 3, 3, 0), [(3, 3)])
    329. 

  329.         self.assertCountEqual(geometry.zone(cell_shape, 3, 3, 1), [(3, 2), (3, 3), (4, 4), (2, 3), (4, 3), (2, 2), (4, 2), (3, 4), (2, 4)])
    330. 

  330.         self.assertCountEqual(geometry.zone(cell_shape, 3, 3, 2), [(2, 4), (3, 2), (5, 4), (1, 3), (4, 5), (2, 1), (1, 4), (2, 3), (4, 2), \
    331. 

  331.                                                                     (5, 1), (2, 5), (3, 5), (5, 3), (1, 2), (3, 3), (5, 5), (4, 4), (3, 1), \
    332. 

  332.                                                                     (1, 5), (4, 3), (2, 2), (4, 1), (5, 2), (3, 4), (1, 1)])
    333. 

  333. 

  334. if __name__ == "__main__":
    335. 

  335.     unittest.main()
    336.