diff options
| -rw-r--r-- | client/requirements.txt | 1 | ||||
| -rw-r--r-- | client/test.py | 5 |
2 files changed, 4 insertions, 2 deletions
diff --git a/client/requirements.txt b/client/requirements.txt index 729cfbb..6ef94a2 100644 --- a/client/requirements.txt +++ b/client/requirements.txt @@ -6,6 +6,7 @@ argparse==1.2.1 beautifulsoup4==4.2.1 facebook-sdk==0.4.0 feedparser==5.1.3 +mock==1.0.1 numpy==1.8.0 python-dateutil==2.1 python-mpd==0.3.0 diff --git a/client/test.py b/client/test.py index a9c5136..7954f8a 100644 --- a/client/test.py +++ b/client/test.py @@ -117,8 +117,9 @@ class TestBrain(unittest.TestCase): def testSortByPriority(self): """Does Brain sort modules by priority?""" my_brain = TestBrain._emptyBrain() - with_priority = filter(lambda m: hasattr(m, 'PRIORITY'), my_brain.modules) - self.assertEqual(sorted(with_priority, key=lambda m: m.PRIORITY), with_priority) + priorities = filter(lambda m: hasattr(m, 'PRIORITY'), my_brain.modules) + target = sorted(priorities, key=lambda m: m.PRIORITY, reverse=True) + self.assertEqual(target, priorities) def testPriority(self): """Does Brain correctly send query to higher-priority module?""" |