Add tests for inifinite classifier

Author: meatballs

axelrod/tests/unit/test_filters.py

@@ -16,6 +16,14 @@ class TestStrategy(object):
             'makes_use_of': ['game', 'length']
         }
 
+    class TestStrategy2(object):
+        classifier = {
+            'stochastic': True,
+            'inspects_source': False,
+            'memory_depth': 'infinity',
+            'makes_use_of': ['game', 'length']
+        }
+
     def test_equality_filter(self):
         self.assertTrue(
             passes_operator_filter(
@@ -30,6 +38,10 @@ def test_equality_filter(self):
             passes_operator_filter(
                 self.TestStrategy, 'inspects_source', True, operator.eq))
 
+        self.assertTrue(
+            passes_operator_filter(
+                self.TestStrategy2, 'memory_depth', float('inf'), operator.eq))
+
     @given(
         smaller=integers(min_value=0, max_value=9),
         larger=integers(min_value=11, max_value=100),
@@ -45,6 +57,9 @@ def test_inequality_filter(self, smaller, larger):
         self.assertFalse(passes_operator_filter(
             self.TestStrategy, 'memory_depth', larger, operator.ge))
 
+        self.assertTrue(passes_operator_filter(
+            self.TestStrategy2, 'memory_depth', smaller, operator.ge))
+
     def test_list_filter(self):
         self.assertTrue(passes_in_list_filter(
             self.TestStrategy, 'makes_use_of', ['game']))