code: def test_augment():
      """
      Test that with 100% punct removing augmentation, the doc never has punct at the end
      """
      args = tagger.parse_args(
          args=["--shorthand", "en_test", "--augment_nopunct", "1.0"]
      )
  
      train_doc = CoNLL.conll2doc(input_str=TRAIN_DATA)
      data = Dataset(train_doc, args, None)
      data = data.to_loader(batch_size=2)
  
      for i in range(50):
          for batch in data:
              for text in batch.text:
                  assert text[-1] not in (".", "!")
  
file_path: stanza/stanza/tests/pos/test_data.py
imports: import os
  import pytest
  from stanza.models.common.doc import *
  from stanza.models import tagger
  from stanza.models.pos.data import Dataset, ShuffledDataset
  from stanza.utils.conll import CoNLL
  from stanza.tests.pos.test_tagger import TRAIN_DATA, TRAIN_DATA_NO_XPOS, TRAIN_DATA_NO_UPOS, TRAIN_DATA_NO_FEATS
label_desc: Use image node_id 6 for calling a global function with example usage: test_augment() without return types