Skip to content

Commit 857c871

Browse files
Irina NicolaeIrina Nicolae
authored andcommitted
Add main func in Gaussian aug unit tests
1 parent 3bf5fbd commit 857c871

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

art/defences/gaussian_augmentation_unittest.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,3 +35,7 @@ def test_labels(self):
3535
self.assertTrue(new_x.shape[0] == new_y.shape[0] == 8)
3636
self.assertTrue(new_x.shape[1:] == x.shape[1:])
3737
self.assertTrue(new_y.shape[1:] == y.shape[1:])
38+
39+
40+
if __name__ == '__main__':
41+
unittest.main()

0 commit comments

Comments
 (0)