def test_targetfield(self): field = TargetField() self.assertTrue(isinstance(field, torchtext.data.Field)) self.assertTrue(field.batch_first) processed = field.preprocessing([None]) self.assertEqual(processed, ['<sos>', None, '<eos>'])
def test_targetfield_with_other_setting(self): field = TargetField(batch_first=False, preprocessing=lambda seq: seq + seq) self.assertTrue(isinstance(field, torchtext.data.Field)) self.assertTrue(field.batch_first) processed = field.preprocessing([None]) self.assertEqual(processed, ['<sos>', None, None, '<eos>'])