/
tests.py
74 lines (58 loc) · 3.02 KB
/
tests.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
# from main import judgeThreeOfAKind, judgeStreak
import unittest
from agents import GreedyAgent, Player
from featureExtractor import FeatureExtractor
from gameState import GameState
# from gameState import judgeThreeOfAKind, judgeStreak
# class TestJudgeMethods(unittest.TestCase):
# def test_judge_threeofakind(self):
# self.assertFalse(judgeThreeOfAKind({"Fire": 0, "Water": 0, "Ice": 0}))
# self.assertTrue(judgeThreeOfAKind({"Fire": 1, "Water": 1, "Ice": 1}))
# self.assertTrue(judgeThreeOfAKind({"Fire": 2, "Water": 2, "Ice": 2}))
# self.assertFalse(judgeThreeOfAKind({"Fire":0, "Water": 2, "Ice": 2}))
# def test_judge_streak(self):
# self.assertFalse(judgeStreak({"Fire": 0, "Water": 0, "Ice": 0}))
# self.assertFalse(judgeStreak({"Fire": 1, "Water": 1, "Ice": 1}))
# self.assertTrue(judgeStreak({"Fire": 3, "Water": 0, "Ice": 0}))
# self.assertTrue(judgeStreak({"Fire": 0, "Water": 3, "Ice": 0}))
# self.assertTrue(judgeStreak({"Fire": 0, "Water": 0, "Ice": 3}))
class TestGreedyAgent(unittest.TestCase):
def pick_highest_num_test(self):
greedyAgent = GreedyAgent("greedy")
greedyAgent.cards = [(2, "Fire"), (3, "Ice"), (4, "Water")]
greedyAgent.pickCard()
self.assertEqual(greedyAgent.playedCard, (4, "Water"))
def pick_low_num_for_streak(self):
greedyAgent = GreedyAgent("greedy")
greedyAgent.cards = [(2, "Fire"), (3, "Ice"), (4, "Water")]
greedyAgent.pickCard()
greedyAgent.accumulatedCards["Water"] += 1
greedyAgent.cards.append((1, "Water"))
greedyAgent.cards.append((0, "Water"))
greedyAgent.pickCard()
self.assertEqual(greedyAgent.playedCard, (1, "Water"))
greedyAgent.accumulatedCards["Water"] += 1
greedyAgent.pickCard()
self.assertEqual(greedyAgent.playedCard, (0, "Water"))
def pick_random(self):
greedyAgent = GreedyAgent("greedy")
greedyAgent.cards = [(3, "Fire"), (3, "Ice"), (3, "Water")]
class TestFeatureExtractor(unittest.TestCase):
def testExtration(self):
featureExt = FeatureExtractor()
agent = Player("aql agent")
enemy = Player("greedy agent")
gameState = GameState(agent, enemy)
enemy.accumulatedCards["Water"] += 1
enemy.accumulatedCards["Fire"] += 1
features = featureExt.getFeatures(gameState, "action", agent.name)
self.assertEqual(features["enemy-distance-to-closest-win"], 1)
self.assertEqual(features["agent-distance-to-closest-win"], 4)
agent.cards.append((1, "Water"))
enemy.accumulatedCards["Fire"] -= 1
enemy.accumulatedCards["Water"] += 1
features = featureExt.getFeatures(gameState, "action", agent.name)
self.assertEqual(features["agent-distance-to-closest-win"], 3)
self.assertEqual(features["enemy-distance-to-closest-win"], 1)
suite = unittest.TestLoader().loadTestsFromTestCase(TestFeatureExtractor)
unittest.TextTestRunner(verbosity=2).run(suite)