98e2f714e54b761c0a1954ad50a26010c7ef50d9,tensorflow_transform/beam/tft_unit.py,TransformTestCase,assertAnalyzeAndTransformResults,#TransformTestCase#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,40

Before Change

    for filename, file_contents in six.iteritems(expected_vocab_file_contents):
      full_filename = tf_transform_output.vocabulary_file_by_name(filename)
      with tf.gfile.Open(full_filename) as f:
        self.assertEqual(f.readlines(), file_contents)

After Change

    for filename, file_contents in six.iteritems(expected_vocab_file_contents):
      full_filename = tf_transform_output.vocabulary_file_by_name(filename)
      with tf.gfile.Open(full_filename) as f:
        file_lines = f.readlines()

        // Store frequency case.
        if isinstance(file_contents[0], tuple):
          word_and_frequency_list = []
          for content in file_lines:
            frequency, word = content.split(" ", 1)
            word_and_frequency_list.append((word.strip("\n"),
                                            float(frequency.strip("\n"))))
          self.assertAllEqual(
              zip(*word_and_frequency_list)[0], zip(*file_contents)[0])
          np.testing.assert_almost_equal(
              zip(*word_and_frequency_list)[1], zip(*file_contents)[1])
        else:
          file_lines = [content.strip("\n") for content in file_lines]
          self.assertAllEqual(file_lines, file_contents)

Instances