diff --git a/tests/test_main.py b/tests/test_main.py index e55e321..5d8ac24 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -64,7 +64,7 @@ def test_parse_args(self): # test error when using --cls with --bins with self.assertRaises(SystemExit): - parse_args(['predict', 'image.jpg', '--cls', 'class1,class2', '--bins', 'somefile.csv', 'genus']) + parse_args(['predict', 'image.jpg', '--cls', 'class1,class2', '--bins', 'somefile.csv']) # not an error when using --cls with --k args = parse_args(['predict', 'image.jpg', '--cls', 'class1,class2', '--k', '10'])