def setUp(self):
ClientTest.setUp(self)
self.client.login(username='******', password='******')
self.source_id = Source.objects.get(name='Labelset 2keys').pk
def setUp(self):
ClientTest.setUp(self)
self.client.login(username='******', password='******')
self.source_id = Source.objects.get(name='Labelset 2keys').pk
# Files to upload.
self.image_filenames = [
os.path.join('2keys', 'cool_001_2011-05-28.png'),
os.path.join('2keys', 'cool_001_2012-05-28.png'),
os.path.join('2keys', 'cool_002_2011-05-28.png'),
]
self.annotation_file_contents_with_labels = (
"cool; 001; 2011; 200; 300; Scarlet\n"
"cool; 001; 2011; 50; 250; Lime\n"
"cool; 001; 2011; 10; 10; Turq\n"
"\n"
"cool; 001; 2012; 1; 1; UMarine\n"
"cool; 001; 2012; 400; 400; Lime\n"
"\n"
"cool; 002; 2011; 160; 40; Turq\n"
"\n"
"will_not_be_uploaded; 025; 2004; 1465; 797; UMarine\n"
)
self.annotation_file_contents_without_labels = (
"cool; 001; 2011; 200; 300\n"
"cool; 001; 2011; 50; 250\n"
"cool; 001; 2011; 10; 10\n"
"\n"
"cool; 001; 2012; 1; 1\n"
"cool; 001; 2012; 400; 400\n"
"\n"
"cool; 002; 2011; 160; 40\n"
"\n"
"will_not_be_uploaded; 025; 2004; 1465; 797\n"
)
# Number of annotations that should be recognized by the annotation
# file check. Note that the annotation file check does not know
# what image files are actually going to be uploaded; so if the
# annotation file contains annotations for image A, then it would
# report an annotation count for image A even if A isn't uploaded.
self.expected_annotations_per_image = {
'cool;001;2011': 3,
'cool;001;2012': 2,
'cool;002;2011': 1,
'will_not_be_uploaded;025;2004': 1,
}
# The annotations that should actually be created after the upload
# completes.
self.expected_annotations = {
'cool_001_2011-05-28.png': set([
(200, 300, 'Scarlet'),
(50, 250, 'Lime'),
(10, 10, 'Turq'),
]),
'cool_001_2012-05-28.png': set([
(1, 1, 'UMarine'),
(400, 400, 'Lime'),
]),
'cool_002_2011-05-28.png': set([
(160, 40, 'Turq'),
]),
}
# Same as expected_annotations, but for the points-only option.
self.expected_points = {
'cool_001_2011-05-28.png': set([
(200, 300),
(50, 250),
(10, 10),
]),
'cool_001_2012-05-28.png': set([
(1, 1),
(400, 400),
]),
'cool_002_2011-05-28.png': set([
(160, 40),
]),
}