def METHOD_NAME(self, create_n):
    for t in self.types:
        if t == "mlm":
            aug = CharInsert(
                "mlm", create_n=create_n, model_name="__internal_testing__/tiny-random-ernie", vocab="test_vocab"
            )
            augmented = aug.augment(self.sequences)
            self.assertEqual(len(self.sequences), len(augmented))
            continue
        elif t == "custom":
            aug = CharInsert(
                "custom", create_n=create_n, custom_file_path=self.custom_file_path, vocab="test_vocab"
            )
        else:
            aug = CharInsert(t, create_n=create_n, vocab="test_vocab")
        augmented = aug.augment(self.sequences)
        self.assertEqual(len(self.sequences), len(augmented))
        self.assertEqual(create_n, len(augmented[0]))
        self.assertEqual(create_n, len(augmented[1]))