Home
last modified time | relevance | path

Searched refs:mock_log (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/keras/utils/
Dio_utils_test.py33 with test.mock.patch.object(six.moves, 'input') as mock_log:
34 mock_log.return_value = 'y'
37 mock_log.return_value = 'n'
41 mock_log.side_effect = ['m', 'y']
44 mock_log.side_effect = ['m', 'n']
/external/tensorflow/tensorflow/python/kernel_tests/
Dsummary_ops_test.py1166 with test.mock.patch.object(logging, 'warn') as mock_log:
1169 str(mock_log.call_args), 'Cannot enable trace inside a tf.function.')
1173 with test.mock.patch.object(logging, 'warn') as mock_log:
1177 str(mock_log.call_args), 'Must enable trace in eager mode.')
1196 with test.mock.patch.object(logging, 'warn') as mock_log:
1199 str(mock_log.call_args), 'Cannot export trace inside a tf.function.')
1203 with test.mock.patch.object(logging, 'warn') as mock_log:
1207 str(mock_log.call_args),
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_dataset_test.py463 with test.mock.patch.object(logging, 'warning') as mock_log:
473 str(mock_log.call_args))
478 'building your dataset.', str(mock_log.call_args))
Dtraining_utils_v1_test.py188 with test.mock.patch.object(logging, 'warning') as mock_log:
191 str(mock_log.call_args), 'input dataset `x` is not shuffled.')
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks_test.py226 def mock_log(*args, **kwargs): function
230 tf_logging.info = mock_log
978 with test.mock.patch.object(tf_logging, 'warning') as mock_log:
983 self.assertIsNone(mock_log.call_args)
1070 with test.mock.patch.object(tf_logging, 'log_first_n') as mock_log:
1074 str(mock_log.call_args), 'global step.*has not been increased')
1423 with test.mock.patch.object(tf_logging, 'warning') as mock_log:
1428 str(mock_log.call_args), 'dependency back to some input source')
Dcheckpoint_management_test.py507 with test.mock.patch.object(logging, "warning") as mock_log:
511 str(mock_log.call_args),
/external/tensorflow/tensorflow/python/keras/tests/
Dadd_loss_correctness_test.py457 with test.mock.patch.object(logging, 'warning') as mock_log:
460 str(mock_log.call_args))
Dtracking_util_test.py383 with test.mock.patch.object(logging, "warning") as mock_log:
387 messages = str(mock_log.call_args_list)
/external/tensorflow/tensorflow/python/training/tracking/
Dutil_test.py348 with test.mock.patch.object(logging, "warning") as mock_log:
356 messages = str(mock_log.call_args_list)
385 with test.mock.patch.object(logging, "warning") as mock_log:
389 self.assertEmpty(mock_log.call_args_list)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Ddataset_test.py218 with test.mock.patch.object(warnings, "warn") as mock_log:
221 self.assertEmpty(mock_log.call_args_list)
/external/tensorflow/tensorflow/python/keras/
Dcallbacks_test.py1319 with test.mock.patch.object(logging, 'warning') as mock_log:
1322 str(mock_log.call_args), '`epsilon` argument is deprecated')
2747 with test.mock.patch.object(logging, 'warn') as mock_log:
2752 str(mock_log.call_args), 'Model failed to serialize as JSON.')
2759 with test.mock.patch.object(logging, 'warn') as mock_log:
2765 str(mock_log.call_args),
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_test.py97 with test.mock.patch.object(logging, "warning") as mock_log:
99 self.assertRegex(str(mock_log.call_args), "already been initialized")