def testText2SelfTxtIterator(self): targets = [ entry["targets"] for entry in text_problems.text2self_txt_iterator( self.targets_file) ] self.assertEqual(targets, self.targets)
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)