Skip to content

Commit

Permalink
fixing test
Browse files Browse the repository at this point in the history
  • Loading branch information
snyaggarwal committed Sep 1, 2023
1 parent 521ae77 commit f5d4b5f
Showing 1 changed file with 5 additions and 13 deletions.
18 changes: 5 additions & 13 deletions core/integration_tests/tests_concepts.py
Original file line number Diff line number Diff line change
Expand Up @@ -1699,31 +1699,23 @@ def test_search(self): # pylint: disable=too-many-statements
HTTP_AUTHORIZATION='Token ' + self.token,
)
self.assertEqual(response.status_code, 200)
self.assertEqual(
response.data,
[
{'name': 'high', 'threshold': ANY, 'confidence': ANY, 'total': ANY},
{'name': 'medium', 'threshold': ANY, 'confidence': ANY, 'total': 0},
{'name': 'low', 'threshold': 0.01, 'confidence': '<50.0%', 'total': 0}
]
)
self.assertTrue(response.data[0]['total'] >= 2)

response = self.client.get(
self.source.uri + 'v1/concepts/?q=MyConcept',
self.source.uri + 'HEAD/concepts/?q=MyConcept',
HTTP_AUTHORIZATION='Token ' + self.token,
)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 2)
self.assertEqual(response.data[0]['id'], 'MyConcept1')
self.assertEqual(response.data[1]['id'], 'MyConcept2')

response = self.client.get(
self.source.uri + 'HEAD/concepts/?q=MyConcept',
self.source.uri + 'v1/concepts/?q=MyConcept',
HTTP_AUTHORIZATION='Token ' + self.token,
)
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.data), 2)
self.assertEqual(response.data[0]['id'], 'MyConcept1')
self.assertEqual(response.data[1]['id'], 'MyConcept2')
self.assertEqual(len(response.data), 1)

def test_facets(self):
if settings.ENV == 'ci':
Expand Down

0 comments on commit f5d4b5f

Please sign in to comment.