Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add missing tests for lookerup.py #900

Merged
merged 7 commits into from
Mar 14, 2017
Merged

Add missing tests for lookerup.py #900

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
11874c7
Add test for exception in lookerup
meatballs Mar 13, 2017
20d6f0c
Add simple test for create_lookup_table_keys
meatballs Mar 13, 2017
81c28c4
Add simple test for create_lookup_table_from_pattern
meatballs Mar 13, 2017
3e9df7a
Add first play test
meatballs Mar 13, 2017
170f18e
Add versus test
meatballs Mar 13, 2017
7b0cb4e
Correct init_kwargs order
meatballs Mar 13, 2017
9e24566
Replace 'C' and 'D' with C and D
meatballs Mar 14, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
57 changes: 45 additions & 12 deletions axelrod/tests/strategies/test_lookerup.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
"""Test for the Looker Up strategy."""
import copy
import unittest

import axelrod
from axelrod.strategies.lookerup import (
Expand Down Expand Up @@ -28,6 +27,31 @@ class TestLookerUp(TestPlayer):
expected_class_classifier = copy.copy(expected_classifier)
expected_class_classifier['memory_depth'] = float('inf')

def test_create_lookup_table_keys(self):
expected = [
(C, C, C), (C, C, D), (C, D, C), (C, D, D),
(D, C, C), (D, C, D), (D, D, C), (D, D, D)
]
actual = create_lookup_table_keys(1, 1, 1)
self.assertEqual(actual, expected)

def test_create_lookup_table_from_pattern(self):
expected = {
(C, C, C): C,
(C, C, D): C,
(C, D, C): D,
(C, D, D): C,
(D, C, C): C,
(D, C, D): D,
(D, D, C): C,
(D, D, D): C
}
actual = create_lookup_table_from_pattern(1, 1, 1, 'CCDCCDCC')
self.assertEqual(actual, expected)

with self.assertRaises(ValueError):
create_lookup_table_from_pattern(2, 2, 2, 'CCC')

def test_init(self):
# Test empty table
player = self.player(dict())
Expand All @@ -36,10 +60,10 @@ def test_init(self):
# Test default table
player = self.player()
expected_lookup_table = {
('', 'C', 'D'): D,
('', 'D', 'D'): D,
('', 'C', 'C'): C,
('', 'D', 'C'): C,
('', C, D): D,
('', D, D): D,
('', C, C): C,
('', D, C): C,
}
self.assertEqual(player.lookup_table, expected_lookup_table)
# Test malformed tables
Expand All @@ -54,18 +78,27 @@ def test_pattern_init(self):
# Test default table
player = self.player(lookup_pattern=pattern, parameters=parameters)
expected_lookup_table = {
('', 'C', 'D'): C,
('', 'D', 'D'): C,
('', 'C', 'C'): C,
('', 'D', 'C'): C,
('', C, D): C,
('', D, D): C,
('', C, C): C,
('', D, C): C,
}
self.assertEqual(player.lookup_table, expected_lookup_table)

def test_strategy(self):
self.first_play_test(C)
self.second_play_test(C, D, C, D) # TFT
self.responses_test([C], [C] * 4, [C, C, C, C])
self.responses_test([D], [C] * 5, [C, C, C, C, D])

opponent = axelrod.MockPlayer()
actions = [(C, C), (C, C), (C, C), (C, C)]
self.versus_test(
opponent,
expected_actions=actions,
init_kwargs={'parameters': (1, 0, 1)}
)

def test_defector_table(self):
"""
Testing a lookup table that always defects if there is enough history.
Expand Down Expand Up @@ -168,9 +201,9 @@ class TestEvolvedLookerUp2_2_2(TestPlayer):

def test_init(self):
# Check for a few known keys
known_pairs = {('DD', 'CC', 'CD'): 'D', ('DC', 'CD', 'CD'): 'C',
('DD', 'CD', 'CD'): 'C', ('DC', 'DC', 'DC'): 'C',
('DD', 'DD', 'CC'): 'D', ('CD', 'CC', 'DC'): 'D'}
known_pairs = {('DD', 'CC', 'CD'): D, ('DC', 'CD', 'CD'): C,
('DD', 'CD', 'CD'): C, ('DC', 'DC', 'DC'): C,
('DD', 'DD', 'CC'): D, ('CD', 'CC', 'DC'): D}
player = self.player()
for k, v in known_pairs.items():
self.assertEqual(player.lookup_table[k], v)
Expand Down