class TestWebServiceClient(object): def setUp(self): self.base_url = 'http://localhost:8084/kripo' self.client = WebserviceClient(self.base_url) @requests_mock.mock() def test_similar_fragments(self, m): expected = [ {'query_frag_id': '3j7u_NDP_frag24', 'hit_frag_id': '3j7u_NDP_frag23', 'score': 0.8991}, ] url = self.base_url + '/fragments/3j7u_NDP_frag24/similar?cutoff=0.75&limit=1' m.get(url, json=expected) response = self.client.similar_fragments(fragment_id='3j7u_NDP_frag24', cutoff=0.75, limit=1) eq_(response, expected)
def client(base_url): return WebserviceClient(base_url)
def setUp(self): self.base_url = 'http://localhost:8084/kripo' self.client = WebserviceClient(self.base_url)