@@ -17,7 +17,7 @@ def tearDown(self):
1717
1818 def test_gpu_check (self ):
1919
20- argv = ['-x' , '-g' , '2' , '-G' , '1' , '-q' , 'gpu_1 ' , 'launch' , 'echo' , 'testing123' ]
20+ argv = ['-x' , '-g' , '2' , '-G' , '1' , '-C' , '1' , '- q' , 'random ' , '-t' , '10' , 'launch' , 'echo' , 'testing123' ]
2121
2222 with self .assertRaises (SystemExit ) as context :
2323 smartdispatch_script .main (argv = argv )
@@ -26,7 +26,7 @@ def test_gpu_check(self):
2626
2727 def test_cpu_check (self ):
2828
29- argv = ['-x' , '-c' , '2' , '-C' , '1' , '-q' , 'gpu_1 ' , 'launch' , 'echo' , 'testing123' ]
29+ argv = ['-x' , '-c' , '2' , '-C' , '1' , '-G' , '1' , '-t' , '10' , '- q' , 'random ' , 'launch' , 'echo' , 'testing123' ]
3030
3131 with self .assertRaises (SystemExit ) as context :
3232 smartdispatch_script .main (argv = argv )
@@ -37,7 +37,7 @@ def test_cpu_check(self):
3737 def test_launch_job_check (self , mock_check_output ):
3838
3939 mock_check_output .side_effect = subprocess .CalledProcessError (1 , 1 , "A wild error appeared!" )
40- argv = ['-q' , 'gpu_1 ' , 'launch' , 'echo' , 'testing123' ]
40+ argv = ['-t' , '0:0:1' , '-G' , '1' , '-C' , '1' , '- q' , 'random ' , 'launch' , 'echo' , 'testing123' ]
4141
4242 try :
4343 with self .assertRaises (SystemExit ) as context :
0 commit comments