|
1 | 1 | import unittest |
2 | 2 |
|
3 | | -from axelrod import filtered_strategies |
4 | | -from axelrod.strategies._filters import * |
5 | | - |
6 | 3 | from hypothesis import example, given, settings |
7 | 4 | from hypothesis.strategies import integers |
8 | 5 |
|
| 6 | +from axelrod import filtered_strategies |
| 7 | +from axelrod.player import Player |
| 8 | +from axelrod.strategies._filters import * |
| 9 | + |
9 | 10 |
|
10 | 11 | class TestFilters(unittest.TestCase): |
11 | | - class TestStrategy(object): |
| 12 | + class TestStrategy(Player): |
12 | 13 | classifier = { |
13 | 14 | "stochastic": True, |
14 | 15 | "inspects_source": False, |
@@ -126,17 +127,17 @@ def test_passes_filterset(self, smaller, larger): |
126 | 127 | self.assertFalse(passes_filterset(self.TestStrategy, sparse_failing_filterset)) |
127 | 128 |
|
128 | 129 | def test_filtered_strategies(self): |
129 | | - class StochasticTestStrategy(object): |
| 130 | + class StochasticTestStrategy(Player): |
130 | 131 | classifier = { |
131 | 132 | "stochastic": True, |
132 | 133 | "memory_depth": float("inf"), |
133 | 134 | "makes_use_of": [], |
134 | 135 | } |
135 | 136 |
|
136 | | - class MemoryDepth2TestStrategy(object): |
| 137 | + class MemoryDepth2TestStrategy(Player): |
137 | 138 | classifier = {"stochastic": False, "memory_depth": 2, "makes_use_of": []} |
138 | 139 |
|
139 | | - class UsesLengthTestStrategy(object): |
| 140 | + class UsesLengthTestStrategy(Player): |
140 | 141 | classifier = { |
141 | 142 | "stochastic": True, |
142 | 143 | "memory_depth": float("inf"), |
|
0 commit comments