def test(self):
    self.assertEqual(self.input_augment_lt.axes, self.input_lt.axes)
    self.assertEqual(self.target_augment_lt.axes, self.target_lt.axes)

    self.save_images('augment', [
        self.get_images('input_', self.input_augment_lt),
        self.get_images('target_', self.target_augment_lt)
    ])
    self.assert_images_near('augment', True)