diff options
Diffstat (limited to 'tensorflow/python/tools/saved_model_cli_test.py')
| -rw-r--r-- | tensorflow/python/tools/saved_model_cli_test.py | 22 |
1 files changed, 0 insertions, 22 deletions
diff --git a/tensorflow/python/tools/saved_model_cli_test.py b/tensorflow/python/tools/saved_model_cli_test.py index eedc893a38..f99c844845 100644 --- a/tensorflow/python/tools/saved_model_cli_test.py +++ b/tensorflow/python/tools/saved_model_cli_test.py @@ -525,28 +525,6 @@ signature_def['serving_default']: y_expected = np.array([[2.5], [3.0]]) self.assertAllClose(y_expected, y_actual) - def testScanCommand(self): - self.parser = saved_model_cli.create_parser() - base_path = test.test_src_dir_path(SAVED_MODEL_PATH) - args = self.parser.parse_args(['scan', '--dir', base_path]) - with captured_output() as (out, _): - saved_model_cli.scan(args) - output = out.getvalue().strip() - self.assertTrue('does not contain blacklisted ops' in output) - - def testScanCommandFoundBlacklistedOp(self): - self.parser = saved_model_cli.create_parser() - base_path = test.test_src_dir_path(SAVED_MODEL_PATH) - args = self.parser.parse_args( - ['scan', '--dir', base_path, '--tag_set', 'serve']) - op_blacklist = saved_model_cli._OP_BLACKLIST - saved_model_cli._OP_BLACKLIST = set(['VariableV2']) - with captured_output() as (out, _): - saved_model_cli.scan(args) - saved_model_cli._OP_BLACKLIST = op_blacklist - output = out.getvalue().strip() - self.assertTrue('\'VariableV2\'' in output) - if __name__ == '__main__': test.main() |