code: def test_augment_punct():
      data = [["Simple", "test", "."]]
      should_augment = lambda x: x[-1] == "."
      can_augment = should_augment
      new_data = augment_punct(data, 1.0, should_augment, can_augment)
      assert new_data == [["Simple", "test"]]
  
imports: import pytest
  import stanza
  from stanza.tests import *
  from stanza.models.common.data import get_augment_ratio, augment_punct
label_desc: Use image node_id 2 for calling a global function with example usage: test_augment_punct() without return types