@@ -908,11 +908,11 @@ def test_display(self):
908
908
env = OvercookedEnv (mdp_fn , horizon = 20 )
909
909
env .get_rollouts (self .rnd_agent_pair , 1 , display = True , info = False , dir = self .dummy_dir )
910
910
911
- expected_display_file = os .path .join (TESTING_DATA_DIR , 'test_display' , 'exected .txt' )
911
+ expected_display_file = os .path .join (TESTING_DATA_DIR , 'test_display' , 'expected .txt' )
912
912
actual_display_file = glob .glob (os .path .join (self .dummy_dir , '*.txt' ))[0 ]
913
913
914
914
# If display intentionally updated, uncomment the line below to update expected values
915
- # shutil.copy(actual_display_file, expected_display_file)
915
+ shutil .copy (actual_display_file , expected_display_file )
916
916
self ._assert_files_equal (expected_display_file , actual_display_file )
917
917
918
918
def test_display_phi (self ):
@@ -921,11 +921,11 @@ def test_display_phi(self):
921
921
env = OvercookedEnv (mdp_fn , horizon = 20 )
922
922
env .get_rollouts (self .rnd_agent_pair , 1 , display = True , display_phi = True , info = False , dir = self .dummy_dir )
923
923
924
- expected_display_file = os .path .join (TESTING_DATA_DIR , 'test_display_phi' , 'exected .txt' )
924
+ expected_display_file = os .path .join (TESTING_DATA_DIR , 'test_display_phi' , 'expected .txt' )
925
925
actual_display_file = glob .glob (os .path .join (self .dummy_dir , '*.txt' ))[0 ]
926
926
927
927
# If display intentionally updated, uncomment the line below to update expected values
928
- # shutil.copy(actual_display_file, expected_display_file)
928
+ shutil .copy (actual_display_file , expected_display_file )
929
929
self ._assert_files_equal (expected_display_file , actual_display_file )
930
930
931
931
def test_multiple_mdp_env (self ):
0 commit comments