generatejson_test.py 文件源码

def test_bad_detection_type(self):
        input_file = StringIO(self.image_name + ' 125:1')
        output_filename = os.path.join(self.temp_dir, 'empty_file.json')
        main(input_file, output_filename)
        self.assertTrue(os.path.exists(output_filename))
        with open(output_filename, 'r') as output_file:
            obj = json.load(output_file)
        self.assertEqual(1, len(obj['requests']))
        self.assertIn('image', obj['requests'][0])
        self.assertIn('content', obj['requests'][0]['image'])
        self.assertIn('features', obj['requests'][0])
        self.assertEqual(1, len(obj['requests'][0]['features']))
        self.assertEqual(
            'TYPE_UNSPECIFIED', obj['requests'][0]['features'][0]['type'])
        self.assertEqual(
            1, obj['requests'][0]['features'][0]['maxResults'])