def setUp(self): taxonomy = Taxonomy('base', {'key': 'value', 'key2': 'value2'}) component1 = Engine('recommender1', taxonomy, {'base_url': 'http://localhost'}) component2 = Engine('recommender2', taxonomy, {'base_url': 'http://localhost2'}) components = {'component1': component1, 'component2': component2} settings = {'test': 'value'} self.engine = HybridEngine('hybrid', taxonomy, components, settings)
def setUp(self): taxonomy = Taxonomy('base', {'key': 'value', 'key2': 'value2'}) self.engine = Engine('recommender1', taxonomy, { 'base_url': 'http://localhost', 'key': 'value' }) self.engine.requests = MagicMock() self.engine.requests.request = MagicMock() self.engine.requests.get = MagicMock()
def setUp(self): taxonomy = Taxonomy('base', {'key': 'value', 'key2': 'value2'}) component1 = Engine('recommender1', taxonomy, { 'base_url': 'http://localhost' }) component1.requests = MagicMock() component1.requests.get = MagicMock() component2 = Engine('recommender2', taxonomy, { 'base_url': 'http://localhost2' }) component2.requests = MagicMock() component2.requests.get = MagicMock() components = {'component1': component1, 'component2': component2} settings = {'weight': {'component1': 0.25, 'component2': 0.75}} self.engine = HybridEngine('hybrid', taxonomy, components, settings)
class EngineTestCase(unittest.TestCase): def setUp(self): taxonomy = Taxonomy('base', {'key': 'value', 'key2': 'value2'}) self.engine = Engine('recommender1', taxonomy, { 'base_url': 'http://localhost', 'key': 'value' }) self.engine.requests = MagicMock() self.engine.requests.request = MagicMock() self.engine.requests.get = MagicMock() def test_translate(self): body = {'value': 'test', 'value3': 'test3'} translated = self.engine.translate(body) self.assertDictEqual(translated, {'key': 'test', 'key2': 'value2'}) def test_post(self): response_mock = MagicMock(status_code=204) self.engine.requests.request.return_value = response_mock self.assertTrue(self.engine.post('/test', {'value': 'test'})) response_mock.status_code = MagicMock(return_value=401) self.assertFalse(self.engine.post('/test', {'value': 'test'})) def test_delete(self): response_mock = MagicMock(status_code=204) self.engine.requests.request.return_value = response_mock self.assertTrue(self.engine.delete('/test', {'value': 'test'})) response_mock.status_code = MagicMock(return_value=401) self.assertFalse(self.engine.delete('/test', {'value': 'test'})) def test_recommend(self): response_mock = MagicMock() response_mock.json = MagicMock(return_value=OrderedDict(( ('result1', 1.5), ('result2', 1) ))) self.engine.requests.get.return_value = response_mock results = self.engine.recommend({'value': 'test'}) self.assertEqual(results, ['result1', 'result2']) results = self.engine.recommend({'value': 'test'}, True) self.assertEqual(results, OrderedDict((('result1', 1.5), ('result2', 1)))) def test_add_brackets_to_lists(self): data = {'test': 'value', 'test2': ['value2', 'value3']} self.assertEqual(self.engine.add_brackets_to_lists(data), { 'test': 'value', 'test2[]': ['value2', 'value3'] })
def setUp(self): taxonomy = Taxonomy('base', {'key': 'value', 'key2': 'value2'}) component1 = Engine('recommender1', taxonomy, {'base_url': 'http://localhost'}) component1.requests = MagicMock() component1.requests.get = MagicMock() component2 = Engine('recommender2', taxonomy, {'base_url': 'http://localhost2'}) component2.requests = MagicMock() component2.requests.get = MagicMock() components = {'component1': component1, 'component2': component2} settings = {'weight': {'component1': 0.25, 'component2': 0.75}} self.engine = HybridEngine('hybrid', taxonomy, components, settings)