test_json_parser_v2.py

    def test_add_frame(self):
        frames_ids = [1, 2, 3, 4, 5]
        self.json_parser.set_top_k(0)
        for frame_id in frames_ids:
            self.json_parser.add_frame(frame_id)
        output = self.json_parser.output()
        output_ids = [frame['frame_id'] for frame in output['frames']]
        for frame_id in frames_ids:
            self.assertIn(frame_id, output_ids)

        # if repeated frame id was inserted, raise ValueError
        with self.assertRaisesRegex(ValueError, "Frame id: (.*?) already exists"):
            self.json_parser.add_frame(frames_ids[1])