diff options
Diffstat (limited to 'tests/test_constraint.py')
-rw-r--r-- | tests/test_constraint.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/tests/test_constraint.py b/tests/test_constraint.py index c92d0b3..438fa97 100644 --- a/tests/test_constraint.py +++ b/tests/test_constraint.py @@ -159,3 +159,28 @@ class TestConstraintManager(unittest.TestCase): self.assertEquals(result[0].iloc[0], 3) self.assertEquals(result[1].iloc[0], 3) + + def test_constraint_with_window(self): + """Test that the constraint manager can constraint to a window of time""" + c_mgr = ConstraintManager(self.dfrs[0], "freq", None, AttrConf.PIVOT, {}, + window=(1, 3)) + + constraint = iter(c_mgr).next() + series = constraint.result[AttrConf.PIVOT_VAL] + self.assertEquals(len(series), 3) + + # For the graph to plot a value at 0.75, the resulting series + # must contain the value before 0.75. Same for the upper limit. + c_mgr = ConstraintManager(self.dfrs[0], "freq", None, AttrConf.PIVOT, {}, + window=(0.75, 1.5)) + + constraint = iter(c_mgr).next() + series = constraint.result[AttrConf.PIVOT_VAL] + self.assertEquals(series.index.tolist(), [0, 1, 2]) + + c_mgr = ConstraintManager(self.dfrs[0], "freq", None, AttrConf.PIVOT, {}, + window=(0, 2)) + + constraint = iter(c_mgr).next() + series = constraint.result[AttrConf.PIVOT_VAL] + self.assertEquals(len(series), 3) |