Esempio n. 1
0
    def test_notes(self):
        # Check default
        do = DetectedObject(self.bbox)
        nt.assert_equal(do.notes, [])

        # Clearing empty is OK
        do.clear_notes()
        nt.assert_equal(do.notes, [])

        # Add a few values
        do.add_note(self.note_to_add)
        exp_notes = [self.note_to_add]
        nt.assert_equal(do.notes, exp_notes)

        new_note = "other_example_note"
        do.add_note(new_note)
        exp_notes.append(new_note)
        nt.assert_equal(do.notes, exp_notes)

        # Clearing works as expected
        do.clear_notes()
        nt.assert_equal(do.notes, [])
Esempio n. 2
0
    def test_clone(self):
        do = DetectedObject(self.bbox)
        do_clone = do.clone()
        nt.ok_(self.check_det_objs_equal(do, do_clone))

        do = DetectedObject(self.bbox, self.conf, self.dot, self.mask)
        do_clone = do.clone()
        nt.ok_(self.check_det_objs_equal(do, do_clone))

        # Try setting some values
        do.geo_point = self.geo_point
        do.index = self.index
        do.detector_name = self.detector_name
        do.set_descriptor(self.descriptor)
        do.add_note(self.note_to_add)

        do.add_keypoint(self.keypoint_id, self.keypoint_to_add)
        # First show that its a deep copy. Should no longer be equal
        nt.assert_false(self.check_det_objs_equal(do, do_clone))

        # Now clone
        do_clone = do.clone()
        nt.ok_(self.check_det_objs_equal(do, do_clone))