test(nyz): fix dreamer unittest bugs

Author: PaParaZz1

ding/world_model/tests/test_dreamer.py renamed to ding/world_model/tests/test_dreamerv3.py

@@ -7,7 +7,7 @@
 from ding.utils import deep_merge_dicts
 
 # arguments
-state_size = [3, 64, 64]
+state_size = [[3, 64, 64]]
 action_size = [6, 1]
 args = list(product(*[state_size, action_size]))
 

ding/world_model/tests/test_world_model.py

@@ -1,4 +1,5 @@
 import pytest
+import os
 import torch
 from easydict import EasyDict
 from ding.world_model.base_world_model import DreamWorldModel, DynaWorldModel
@@ -10,8 +11,8 @@ class TestDynaWorldModel:
 
     @pytest.mark.parametrize('buffer_type', [NaiveReplayBuffer, EpisodeReplayBuffer])
     def test_fill_img_buffer(self, buffer_type):
-        env_buffer = buffer_type(buffer_type.default_config(), None, 'exp_name', 'env_buffer_for_test')
-        img_buffer = buffer_type(buffer_type.default_config(), None, 'exp_name', 'img_buffer_for_test')
+        env_buffer = buffer_type(buffer_type.default_config(), None, 'dyna_exp_name', 'env_buffer_for_test')
+        img_buffer = buffer_type(buffer_type.default_config(), None, 'dyna_exp_name', 'img_buffer_for_test')
         fake_config = EasyDict(
             train_freq=250,  # w.r.t environment step
             eval_freq=250,  # w.r.t environment step
@@ -74,6 +75,7 @@ def step(self, obs, action):
         )
 
         super(FakeModel, fake_model).fill_img_buffer(policy, env_buffer, img_buffer, 0, 0)
+        os.popen("rm -rf dyna_exp_name")
 
 
 @pytest.mark.unittest