def test_golden_api_doc(self):
golden = api_gen.load_golden('golden_api.json')
golden_doc = api_gen.load_golden('golden_api_doc.json')
api_keys = list(golden.keys())
doc_keys = list(golden_doc.keys())
msg = ('Expect package keys are matched: \n'
'In `golden_api.json`: \n{}\n\n'
'In `golden_api_doc.json`: \n{}\n\n').format(
api_keys, doc_keys)
self.assertListEqual(api_keys, doc_keys, msg)
def test_golden_api(self):
golden = api_gen.load_golden('golden_api.json')
imports = api_util.generate_imports(api_util.PACKAGE_PREFIX)
imports_json = json.dumps(imports, indent=2, sort_keys=True)
golden_content = api_gen._read_golden_text('golden_api.json')
msg = ('Exported APIs do not match `golden_api.json`. Please check it.\n\n'
'Imports in json format: \n{}\n'
'Golden file content:\n{}\n\n').format(imports_json, golden_content)
self.assertDictEqual(imports, golden, msg)
def test_load_golden(self):
golden = api_gen.load_golden('golden_api.json')
imports = api_util.generate_imports(api_util.PACKAGE_PREFIX)
self.assertDictEqual(imports, golden)
Run this test, after `tflite_model_maker` package is installed.
python golden_api_test.py
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import importlib
from typing import List
from absl.testing import parameterized
import tensorflow as tf
from tensorflow_examples.lite.model_maker.core.api import api_gen
GOLDEN = api_gen.load_golden(api_gen.DEFAULT_API_FILE)
class GoldenApiTest(tf.test.TestCase, parameterized.TestCase):
@parameterized.parameters(GOLDEN.items())
def test_golden_apis(self, package: str, import_lines: List[str]):
"""Test all golden API symbols."""
import tflite_model_maker # pylint: disable=g-import-not-at-top
for line in import_lines:
# Get `question_answer`, for `tflite_model_maker.question_answer`.
parts = package.split('.')[1:]
# Get `c`, for `from a import c` or `from a import b as c`.
name = line.split()[-1]
parts.append(name)
Run this test, after `tflite_model_maker` package is installed.
python golden_api_test.py
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import importlib
from typing import List
from absl.testing import parameterized
import tensorflow as tf
from tensorflow_examples.lite.model_maker.core.api import api_gen
GOLDEN = api_gen.load_golden('golden_api.json')
class GoldenApiTest(tf.test.TestCase, parameterized.TestCase):
@parameterized.parameters(GOLDEN.items())
def test_golden_apis(self, package: str, import_lines: List[str]):
"""Test all golden API symbols."""
import tflite_model_maker # pylint: disable=g-import-not-at-top
for line in import_lines:
if not package:
continue
parts = package.split('.')
# Get `c`, for `from a import c` or `from a import b as c`.
name = line.split()[-1]