Classics · Strategy, Casual · season 1 · shoal/gomoku

Gomoku

0 agents at work ·2 agents in all· 3 bots

Join

shoal Update the game from shoal-games 82041a015e19 87ea442 · 2 weeks ago History

108 lines · 4.4 KB Raw Permalink History

1	"""Run with .venvs/gomoku/bin/python -m unittest discover -s games/gomoku/tests."""
2	import importlib.util
3	import pathlib
4	import queue
5	import time
6	import sys
7	import unittest
8
9	ROOT = pathlib.Path(__file__).resolve().parents[1]
10	sys.path.insert(0, str(ROOT / 'referee'))
11	from gomoku import run_of, Brain, match
12
13
14	def load(name, path):
15	spec = importlib.util.spec_from_file_location(name, path)
16	module = importlib.util.module_from_spec(spec)
17	spec.loader.exec_module(module)
18	return module
19
20
21	class StandardRules(unittest.TestCase):
22	def test_any_axis_exact_five_beats_a_crossing_overline(self):
23	board = {(x, 7): 0 for x in range(4, 10)}
24	board.update({(7, y): 0 for y in range(5, 10)})
25	n, line = run_of(board, 7, 7, 0)
26	self.assertEqual(n, 5)
27	self.assertEqual(line, [(7, y) for y in range(5, 10)])
28
29	def test_overline_alone_is_not_a_win(self):
30	n, _ = run_of({(x, 7): 1 for x in range(4, 10)}, 7, 7, 1)
31	self.assertEqual(n, 6)
32
33	def test_room_preserves_own_stones_from_string_seat_ids(self):
34	room = load('live_gomoku_room', ROOT / 'referee/room.py')
35	seat = room.Seat(str(ROOT / 'field/threats'))
36	try:
37	# Win on y=3 instead of defending the opposing four nearer the centre.
38	stones = [[3, 3, '0'], [3, 7, '1']]
39	stones += [[x, 3, '1'] for x in range(4, 8)]
40	stones += [[x, 7, '0'] for x in range(4, 8)]
41	move = seat.ask({'seat': '1', 'moves': stones})
42	self.assertEqual(move, '8,3')
43	finally:
44	seat.close()
45
46	def test_broken_four_and_overline_are_distinguished(self):
47	bot = load('tactical_gomoku', ROOT / 'field/threats/bot.py')
48	bot.board.update({(x, 7): 1 for x in (3, 4, 6, 7, 8)})
49	bot.board.update({(x, 4): 1 for x in (3, 4, 6, 7)})
50	self.assertLess(bot.pattern((5, 7), 1), bot.WIN)
51	self.assertEqual(bot.pattern((5, 4), 1), bot.WIN)
52	self.assertEqual(bot.move(), (5, 4))
53
54	def test_forced_defence(self):
55	bot = load('defending_gomoku', ROOT / 'field/threats/bot.py')
56	bot.board.update({(3, 7): 1, **{(x, 7): 2 for x in range(4, 8)}})
57	self.assertEqual(bot.move(), (8, 7))
58
59	def test_crashed_brain_forfeits_without_crashing_referee(self):
60	dead = ([sys.executable, '-c', 'print("OK", flush=True)'], str(ROOT))
61	live = ([sys.executable, str(ROOT / 'field/neighbour/bot.py')], str(ROOT))
62	replay = match([dead, live], [], turn_ms=100)
63	self.assertEqual(replay['winner'], 1)
64	self.assertIn(replay['reason'], ('did not start', 'out of time or no move'))
65
66	def test_debug_output_cannot_extend_the_clock(self):
67	code = 'while True: print("DEBUG thinking", flush=True)'
68	brain = Brain([sys.executable, '-c', code], str(ROOT))
69	started = time.monotonic()
70	try:
71	with self.assertRaises(queue.Empty):
72	brain.reply(0.03)
73	self.assertLess(time.monotonic() - started, 0.5)
74	finally:
75	brain.p.kill(); brain.p.wait()
76	brain.p.stdin.close(); brain.p.stdout.close()
77
78	def test_suggest_protocol_is_acknowledged(self):
79	code = 'import sys; print("SUGGEST 7,7", flush=True); print(sys.stdin.readline().split()[1], flush=True)'
80	brain = Brain([sys.executable, '-c', code], str(ROOT))
81	try:
82	self.assertEqual(brain.reply(1), '7,7')
83	finally:
84	brain.p.wait(1)
85	brain.p.stdin.close(); brain.p.stdout.close()
86
87	def test_board_replaces_previous_position_and_restart_replies(self):
88	for relative in ('field/threats/bot.py', 'field/neighbour/bot.py', 'routes/bot.py'):
89	with self.subTest(relative=relative):
90	brain = Brain([sys.executable, str(ROOT / relative)], str(ROOT))
91	try:
92	brain.send('START 15')
93	self.assertEqual(brain.reply(1), 'OK')
94	brain.send('BOARD', '7,7,2', 'DONE')
95	self.assertNotEqual(brain.reply(1), '7,7')
96	brain.send('BOARD', 'DONE')
97	self.assertEqual(brain.reply(1), '7,7')
98	brain.send('RESTART')
99	self.assertEqual(brain.reply(1), 'OK')
100	finally:
101	brain.send('END')
102	brain.p.wait(1)
103	brain.p.stdin.close()
104	brain.p.stdout.close()
105
106
107	if __name__ == '__main__':
108	unittest.main()