diff options
Diffstat (limited to 'tensorflow/tools/api/generator/create_python_api_test.py')
-rw-r--r-- | tensorflow/tools/api/generator/create_python_api_test.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/tools/api/generator/create_python_api_test.py b/tensorflow/tools/api/generator/create_python_api_test.py index 651ec9d040..1a7187463a 100644 --- a/tensorflow/tools/api/generator/create_python_api_test.py +++ b/tensorflow/tools/api/generator/create_python_api_test.py @@ -58,6 +58,7 @@ class CreatePythonApiTest(test.TestCase): def testFunctionImportIsAdded(self): imports = create_python_api.get_api_init_text( package=create_python_api._DEFAULT_PACKAGE, + output_package='tensorflow', api_name='tensorflow') expected_import = ( 'from tensorflow.python.test_module ' @@ -75,6 +76,7 @@ class CreatePythonApiTest(test.TestCase): def testClassImportIsAdded(self): imports = create_python_api.get_api_init_text( package=create_python_api._DEFAULT_PACKAGE, + output_package='tensorflow', api_name='tensorflow') expected_import = ('from tensorflow.python.test_module ' 'import TestClass') @@ -85,6 +87,7 @@ class CreatePythonApiTest(test.TestCase): def testConstantIsAdded(self): imports = create_python_api.get_api_init_text( package=create_python_api._DEFAULT_PACKAGE, + output_package='tensorflow', api_name='tensorflow') expected = ('from tensorflow.python.test_module ' 'import _TEST_CONSTANT') |