Python assert_not_shuffled Beispiele

Beispiel #1

Datei: training_utils_test.py Projekt: Harryi0/tinyML

    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)

Beispiel #2

Datei: training_utils_test.py Projekt: Wajih-O/tensorflow

  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)

Beispiel #3

Datei: training_utils_test.py Projekt: yuyangchen0122/Capstone_2019

 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())