|
| 1 | +from .server import Server, TacticFailure, ServerError |
| 2 | +from .expr import Expr, Tactic, GoalState, Site |
| 3 | + |
1 | 4 | import random |
2 | 5 | from abc import abstractmethod |
3 | 6 | import time |
4 | 7 | from dataclasses import dataclass |
5 | 8 | from typing import Optional, List |
6 | | -import collections, unittest |
| 9 | +import collections |
7 | 10 | from math import log, sqrt |
8 | | -from pantograph.server import Server, TacticFailure, ServerError |
9 | | -from pantograph.expr import Expr, Tactic, GoalState, Site |
| 11 | + |
10 | 12 |
|
11 | 13 |
|
12 | 14 | @dataclass |
@@ -446,55 +448,3 @@ def next_tactic( |
446 | 448 | return tactics[i] |
447 | 449 |
|
448 | 450 |
|
449 | | -class TestSearch(unittest.TestCase): |
450 | | - |
451 | | - def test_solve(self): |
452 | | - |
453 | | - server = Server() |
454 | | - agent = DumbAgent() |
455 | | - goal_state = server.goal_start("∀ (p q: Prop), p -> p") |
456 | | - flag = agent.search( |
457 | | - server=server, |
458 | | - goal_state=goal_state, |
459 | | - verbose=False) |
460 | | - #flag = agent.search(server=server, target="∀ (p q: Prop), Or p q -> Or q p", verbose=True) |
461 | | - self.assertTrue(flag) |
462 | | - def test_solve_big(self): |
463 | | - |
464 | | - server = Server() |
465 | | - agent = DumbAgent() |
466 | | - goal_state = server.goal_start("∀ (p q: Prop), Or p q -> Or q p") |
467 | | - flag = agent.search( |
468 | | - server=server, |
469 | | - goal_state=goal_state, |
470 | | - verbose=False) |
471 | | - self.assertTrue(flag) |
472 | | - |
473 | | -class TestMCTSSearch(unittest.TestCase): |
474 | | - |
475 | | - def test_solve(self): |
476 | | - |
477 | | - server = Server() |
478 | | - agent = DumbMCTSAgent() |
479 | | - goal_state = server.goal_start("∀ (p q: Prop), p -> p") |
480 | | - flag = agent.search( |
481 | | - server=server, |
482 | | - goal_state=goal_state, |
483 | | - verbose=False) |
484 | | - #flag = agent.search(server=server, target="∀ (p q: Prop), Or p q -> Or q p", verbose=True) |
485 | | - self.assertTrue(flag) |
486 | | - def test_solve_big(self): |
487 | | - |
488 | | - server = Server() |
489 | | - agent = DumbMCTSAgent() |
490 | | - goal_state = server.goal_start("∀ (p q: Prop), Or p q -> Or q p") |
491 | | - flag = agent.search( |
492 | | - server=server, |
493 | | - goal_state=goal_state, |
494 | | - max_steps=200, |
495 | | - verbose=False) |
496 | | - self.assertTrue(flag) |
497 | | - |
498 | | - |
499 | | -if __name__ == '__main__': |
500 | | - unittest.main() |
0 commit comments