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, [])
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))