@@ -0,0 +1,1174 @@
+# coding=utf-8%0A# Copyright 2019 The Google Research Authors.%0A#%0A# Licensed under the Apache License, Version 2.0 (the %22License%22);%0A# you may not use this file except in compliance with the License.%0A# You may obtain a copy of the License at%0A#%0A#     http://www.apache.org/licenses/LICENSE-2.0%0A#%0A# Unless required by applicable law or agreed to in writing, software%0A# distributed under the License is distributed on an %22AS IS%22 BASIS,%0A# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.%0A# See the License for the specific language governing permissions and%0A# limitations under the License.%0A%0Afrom __future__ import absolute_import%0Afrom __future__ import division%0Afrom __future__ import print_function%0A%0Aimport run_classifier%0A%0Aimport tensorflow as tf%0A%0A%0Aclass RunClassifierTest(tf.test.TestCase):%0A%0A  def test_step_calculation(self):%0A    num_examples = 10%0A    batch_size = 2%0A    num_epochs = 3%0A    warmup_proportion = 0.5%0A    steps, warmup_steps = run_classifier._calculate_steps(%0A        num_examples, batch_size, num_epochs, warmup_proportion)%0A    self.assertEqual(steps, 15)%0A    self.assertEqual(warmup_steps, 7)%0A%0A%0Aif __name__ == '__main__':%0A  tf.test.main()%0A

Create run_classifier_test.py