def setUp(self): # For these tests we create only one sound and one candidate annotation add_taxonomy_nodes(models.Taxonomy.objects.get()) create_sounds('fsd', 1) create_users(2) create_candidate_annotations('fsd', 1) self.users = models.User.objects.all() self.dataset = models.Dataset.objects.get(short_name='fsd') self.candidate_annotation = models.CandidateAnnotation.objects.first()
def setUp(self): add_taxonomy_nodes(Taxonomy.objects.get()) create_sounds('fsd', 10) create_users(5) create_candidate_annotations('fsd', 20) # add preview url for sounds sounds = Sound.objects.all() with transaction.atomic(): for sound in sounds: sound.extra_data['previews'] = 'http://www.freesound.org/data/previews/188/188440_3399958-hq.ogg' sound.save() self.node_with_candidates = CandidateAnnotation.objects.first().taxonomy_node self.client.login(username='******', password='******')
def setUp(self): # For these tests we create only one sound and one candidate annotation add_taxonomy_nodes(models.Taxonomy.objects.get()) create_sounds('fsd', 1) create_users(2) create_candidate_annotations('fsd', 1) candidate_annotation = models.CandidateAnnotation.objects.first() users = models.User.objects.all() self.dataset = models.Dataset.objects.get(short_name='fsd') # create 2 votes for the candidate annotations (it should create ground truth annotations) for user in users: self.dataset.maintainers.add(user) models.Vote.objects.create( created_by=user, vote=1.0, visited_sound=False, candidate_annotation_id=candidate_annotation.id, test='AP', from_test_page=True, from_task='AD', ) self.dataset.save()
def setUp(self): add_taxonomy_nodes(Taxonomy.objects.get()) create_sounds('fsd', 1) create_users(1) create_candidate_annotations('fsd', 1)
def setUp(self): taxonomy = { "1": { "id": "1", "name": "1", "description": "This is a root category", "child_ids": ["2"], "restrictions": [], "citation_uri": "", }, "2": { "id": "2", "name": "2", "description": "This is a simple non-root & non-leaf category", "child_ids": ["3"], "parent_ids": ["1"], "propagate_to_parent_ids": ["1"], "restrictions": [], "citation_uri": "", }, "3": { "id": "3", "name": "3", "description": "This category has two parents", "child_ids": [], "parent_ids": ["2", "5"], "propagate_to_parent_ids": ["5"], "restrictions": [], "citation_uri": "", }, "4": { "id": "4", "name": "4", "description": "This is a leaf category", "child_ids": [], "parent_ids": ["5"], "propagate_to_parent_ids": ["5"], "restrictions": [], "citation_uri": "", }, "5": { "id": "5", "name": "5", "description": "This is a root category", "child_ids": ["3", "4"], "restrictions": [], "citation_uri": "", }, } self.taxobj = models.Taxonomy.objects.create(data=taxonomy) add_taxonomy_nodes(self.taxobj) models.Dataset.objects.create(short_name='fsd', taxonomy=self.taxobj) create_sounds('fsd', 1) create_users(1) self.sound = models.Sound.objects.first() self.user = models.User.objects.first() self.taxonomy_node = self.taxobj.get_element_at_id( "3") # a children node self.second_taxonomy_node = self.taxobj.get_element_at_id( "4") # a children node # a root node which is propagated from the two nodes above self.taxonomy_node_to_propagate_to = self.taxobj.get_element_at_id("5") create_candidate_annotations('fsd', 1) self.candidate_annotation = models.CandidateAnnotation.objects.create( sound_dataset=self.sound.sounddataset_set.first(), type='AU', algorithm='Fake algorithm name', taxonomy_node=self.taxonomy_node, ground_truth=1.0, ) self.users = models.User.objects.all() self.dataset = models.Dataset.objects.get(short_name='fsd') # self.candidate_annotation = models.CandidateAnnotation.objects.first() # self.candidate_annotation.ground_truth = 1.0 self.candidate_annotation.save() self.gt_annotation = models.GroundTruthAnnotation.objects.create( ground_truth=1.0, created_by=self.user, sound_dataset=self.sound.sounddataset_set.first(), taxonomy_node=self.taxonomy_node) self.gt_annotation.from_candidate_annotations.add( self.candidate_annotation)