# Tests for augmentation module