Skip to content
Snippets Groups Projects
Commit c34075d1 authored by Valentin Rigal's avatar Valentin Rigal Committed by Erwan Rouchet
Browse files

Make ListDatasetElements test independent of the fixture

parent fe0fc10b
No related branches found
No related tags found
1 merge request!2147Make ListDatasetElements test independent of the fixture
Hide whitespace changes
Inline Side-by-side
arkindex/training/tests/test_datasets_api.py
+ 114
− 116
View file @ c34075d1
......@@ -1049,13 +1049,9 @@ class TestDatasetsAPI(FixtureAPITestCase):
self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
def test_list_elements(self):
# id: ddc6a864-82d3-4f4d-9078-6fb547db7279
self.dataset.dataset_elements.create(element_id=self.vol.id, set="test")
# id: ebb0adee-91ff-4d21-bdfd-9dee5e61cd84
self.dataset.dataset_elements.create(element_id=self.page1.id, set="training")
# id: 1a55da30-0c90-4330-9533-02107aaaa93e
self.dataset.dataset_elements.create(element_id=self.page2.id, set="validation")
# id: b7b53865-24f3-4e09-b61c-de8bb1fcbd01
self.dataset.dataset_elements.create(element_id=self.page3.id, set="validation")
self.page3.confidence = 0.42
self.page3.mirrored = True
......@@ -1072,127 +1068,129 @@ class TestDatasetsAPI(FixtureAPITestCase):
data = response.json()
self.assertTrue('?cursor=' in data['next'])
self.assertIsNone(data['count'])
self.maxDiff = None
self.assertListEqual(data['results'], [{
'set': 'validation',
'element': {
'id': str(self.page2.id),
'confidence': None,
'type': 'page',
'mirrored': False,
'corpus': {
'id': str(self.corpus.id),
'name': 'Unit Tests',
'public': True
},
'name': self.page2.name,
'zone': {
self.assertListEqual(data['results'], sorted(
[{
'set': 'validation',
'element': {
'id': str(self.page2.id),
'image': {
'id': str(self.page2.image.id),
'height': 1000,
'width': 1000,
'url': 'http://server/img2',
'path': 'img2',
'status': 'unchecked',
's3_url': None,
'server': {
'display_name': 'Test Server',
'max_height': None,
'max_width': None,
'url': 'http://server'
'confidence': None,
'type': 'page',
'mirrored': False,
'corpus': {
'id': str(self.corpus.id),
'name': 'Unit Tests',
'public': True
},
'name': self.page2.name,
'zone': {
'id': str(self.page2.id),
'image': {
'id': str(self.page2.image.id),
'height': 1000,
'width': 1000,
'url': 'http://server/img2',
'path': 'img2',
'status': 'unchecked',
's3_url': None,
'server': {
'display_name': 'Test Server',
'max_height': None,
'max_width': None,
'url': 'http://server'
},
},
'polygon': [
[0, 0],
[0, 1000],
[1000, 1000],
[1000, 0],
[0, 0]
],
'url': 'http://server/img2/0,0,1000,1000/full/0/default.jpg'
},
'polygon': [
[0, 0],
[0, 1000],
[1000, 1000],
[1000, 0],
[0, 0]
],
'url': 'http://server/img2/0,0,1000,1000/full/0/default.jpg'
'rotation_angle': 0,
'classes': None,
'has_children': None,
'metadata': None,
'worker_run': None,
'worker_version_id': None,
'thumbnail_url': None,
'created': FAKE_CREATED
},
'rotation_angle': 0,
'classes': None,
'has_children': None,
'metadata': None,
'worker_run': None,
'worker_version_id': None,
'thumbnail_url': None,
'created': FAKE_CREATED
},
}, {
'set': 'validation',
'element': {
'id': str(self.page3.id),
'confidence': 0.42,
'type': 'page',
'mirrored': True,
'corpus': {
'id': str(self.corpus.id),
'name': 'Unit Tests',
'public': True
},
'name': self.page3.name,
'zone': {
}, {
'set': 'validation',
'element': {
'id': str(self.page3.id),
'image': {
'id': str(self.page3.image.id),
'height': 1000,
'width': 1000,
'url': 'http://server/img3',
'path': 'img3',
'status': 'unchecked',
's3_url': None,
'server': {
'display_name': 'Test Server',
'max_height': None,
'max_width': None,
'url': 'http://server'
'confidence': 0.42,
'type': 'page',
'mirrored': True,
'corpus': {
'id': str(self.corpus.id),
'name': 'Unit Tests',
'public': True
},
'name': self.page3.name,
'zone': {
'id': str(self.page3.id),
'image': {
'id': str(self.page3.image.id),
'height': 1000,
'width': 1000,
'url': 'http://server/img3',
'path': 'img3',
'status': 'unchecked',
's3_url': None,
'server': {
'display_name': 'Test Server',
'max_height': None,
'max_width': None,
'url': 'http://server'
},
},
'polygon': [
[0, 0],
[0, 1000],
[1000, 1000],
[1000, 0],
[0, 0]
],
'url': 'http://server/img3/0,0,1000,1000/full/0/default.jpg'
},
'polygon': [
[0, 0],
[0, 1000],
[1000, 1000],
[1000, 0],
[0, 0]
],
'url': 'http://server/img3/0,0,1000,1000/full/0/default.jpg'
'rotation_angle': 42,
'classes': None,
'has_children': None,
'metadata': None,
'worker_run': None,
'worker_version_id': None,
'thumbnail_url': None,
'created': FAKE_CREATED
},
'rotation_angle': 42,
'classes': None,
'has_children': None,
'metadata': None,
'worker_run': None,
'worker_version_id': None,
'thumbnail_url': None,
'created': FAKE_CREATED
},
}, {
'set': 'test',
'element': {
'id': str(self.vol.id),
'confidence': None,
'type': 'volume',
'mirrored': False,
'name': 'Volume 1',
'corpus': {
'id': str(self.corpus.id),
'name': 'Unit Tests',
'public': True
}, {
'set': 'test',
'element': {
'id': str(self.vol.id),
'confidence': None,
'type': 'volume',
'mirrored': False,
'name': 'Volume 1',
'corpus': {
'id': str(self.corpus.id),
'name': 'Unit Tests',
'public': True
},
'zone': None,
'rotation_angle': 0,
'classes': None,
'has_children': None,
'metadata': None,
'worker_run': None,
'worker_version_id': None,
'thumbnail_url': self.vol.thumbnail.s3_url,
'created': FAKE_CREATED
},
'zone': None,
'rotation_angle': 0,
'classes': None,
'has_children': None,
'metadata': None,
'worker_run': None,
'worker_version_id': None,
'thumbnail_url': self.vol.thumbnail.s3_url,
'created': FAKE_CREATED
},
}, ])
}],
key=lambda dataset_elt: dataset_elt['element']['id']
))
def test_add_from_selection_requires_login(self):
with self.assertNumQueries(0):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment