def test_verify_dataset_shuffled(self): dataset = dataset_ops.Dataset.range(5) training_utils.assert_not_shuffled(dataset) with test.mock.patch.object(logging, 'warning') as mock_log: training_utils.verify_dataset_shuffled(dataset) self.assertRegexpMatches(str(mock_log.call_args), 'input dataset `x` is not shuffled.') shuffled_dataset = dataset.shuffle(10) training_utils.verify_dataset_shuffled(shuffled_dataset)
def test_verify_dataset_shuffled(self): dataset = dataset_ops.Dataset.range(5) training_utils.assert_not_shuffled(dataset) with test.mock.patch.object(logging, 'warning') as mock_log: training_utils.verify_dataset_shuffled(dataset) self.assertRegexpMatches( str(mock_log.call_args), 'input dataset `x` is not shuffled.') shuffled_dataset = dataset.shuffle(10) training_utils.verify_dataset_shuffled(shuffled_dataset)
def test_assert_not_shuffled(self, dataset_fn, expected_error=None): if expected_error is None: training_utils.assert_not_shuffled(dataset_fn()) else: with self.assertRaises(expected_error): training_utils.assert_not_shuffled(dataset_fn())