File tree Expand file tree Collapse file tree 1 file changed +9
-9
lines changed Expand file tree Collapse file tree 1 file changed +9
-9
lines changed Original file line number Diff line number Diff line change 3535import  unittest 
3636import  subprocess 
3737import  os 
38- from  .autotuner_test_utils  import  AutoTunerTestUtils 
38+ from  .autotuner_test_utils  import  AutoTunerTestUtils ,  accepted_rc 
3939
4040cur_dir  =  os .path .dirname (os .path .abspath (__file__ ))
4141
@@ -75,15 +75,15 @@ def test(self):
7575        if  not  (self .platform  and  self .design ):
7676            raise  unittest .SkipTest ("Platform and design have to be defined" )
7777
78-         tune_process  =  subprocess .Popen (self .tune_command , shell = True )
79-         sweep_process  =  subprocess .Popen (self .sweep_command , shell = True )
80- 
81-         tune_returncode  =  tune_process .wait ()
82-         sweep_returncode  =  sweep_process .wait ()
83- 
84-         self .assertEqual (tune_returncode , 0 , "Tune command failed" )
85-         self .assertEqual (sweep_returncode , 0 , "Sweep command failed" )
78+         # Run tune command 
79+         tune_result  =  subprocess .run (self .tune_command , shell = True )
80+         tune_successful  =  tune_result .returncode  in  accepted_rc 
81+         self .assertTrue (tune_successful )
8682
83+         # Run sweep command 
84+         sweep_result  =  subprocess .run (self .sweep_command , shell = True )
85+         sweep_successful  =  sweep_result .returncode  in  accepted_rc 
86+         self .assertTrue (sweep_successful )
8787
8888
8989if  __name__  ==  "__main__" :
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments