Example #1
0
 def testText2SelfTxtIterator(self):
     targets = [
         entry["targets"] for entry in text_problems.text2self_txt_iterator(
             self.targets_file)
     ]
     self.assertEqual(targets, self.targets)
Example #2
0
    def generate_samples(self, data_dir, _tmp_dir, dataset_split):  #pylint: disable=no-self-use
        docstrings_file_path = os.path.join(
            data_dir, '{}.docstring'.format(dataset_split))

        return text_problems.text2self_txt_iterator(docstrings_file_path)
 def testText2SelfTxtIterator(self):
   targets = [
       entry["targets"]
       for entry in text_problems.text2self_txt_iterator(self.targets_file)
   ]
   self.assertEqual(targets, self.targets)