1from __future__ import absolute_import
2from __future__ import print_function
3
4import os
5import sys
6import six
7from conversion_imagenet import TestModels
8
9def get_test_table():
10    return {
11        'keras' : {
12            'resnet50'     : [
13                TestModels.onnx_emit,
14                TestModels.caffe_emit,
15                TestModels.cntk_emit,
16                TestModels.coreml_emit,
17                TestModels.keras_emit,
18                TestModels.mxnet_emit,
19                TestModels.pytorch_emit,
20                TestModels.tensorflow_emit
21                ]
22    }}
23
24
25def test_keras():
26    test_table = get_test_table()
27    tester = TestModels(test_table)
28    tester._test_function('keras', tester.keras_parse)
29
30
31if __name__ == '__main__':
32    test_keras()
33