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