Skip to content

Commit 0cdabd4

Browse files
committed
Fix non-strategy "strategies" in test_filters
1 parent 09e48cb commit 0cdabd4

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

axelrod/tests/unit/test_filters.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,15 @@
11
import unittest
22

3-
from axelrod import filtered_strategies
4-
from axelrod.strategies._filters import *
5-
63
from hypothesis import example, given, settings
74
from hypothesis.strategies import integers
85

6+
from axelrod import filtered_strategies
7+
from axelrod.player import Player
8+
from axelrod.strategies._filters import *
9+
910

1011
class TestFilters(unittest.TestCase):
11-
class TestStrategy(object):
12+
class TestStrategy(Player):
1213
classifier = {
1314
"stochastic": True,
1415
"inspects_source": False,
@@ -126,17 +127,17 @@ def test_passes_filterset(self, smaller, larger):
126127
self.assertFalse(passes_filterset(self.TestStrategy, sparse_failing_filterset))
127128

128129
def test_filtered_strategies(self):
129-
class StochasticTestStrategy(object):
130+
class StochasticTestStrategy(Player):
130131
classifier = {
131132
"stochastic": True,
132133
"memory_depth": float("inf"),
133134
"makes_use_of": [],
134135
}
135136

136-
class MemoryDepth2TestStrategy(object):
137+
class MemoryDepth2TestStrategy(Player):
137138
classifier = {"stochastic": False, "memory_depth": 2, "makes_use_of": []}
138139

139-
class UsesLengthTestStrategy(object):
140+
class UsesLengthTestStrategy(Player):
140141
classifier = {
141142
"stochastic": True,
142143
"memory_depth": float("inf"),

0 commit comments

Comments
 (0)