test(scene_mapping): Update tests

Author: mikkelkp

tests/scene_mapping_test.py

@@ -5,28 +5,39 @@
 from honeybee_radiance_folder.folderutil import _nukedir
 
 
-def test_scene_mapping():
+def test_scene_mapping_2():
     folder_path = r'./tests/assets/model_folders/simple/model'
     model_folder = Folder.from_model_folder(folder_path)
 
     # two phase
-    scene_mapping = model_folder.octree_scene_mapping(exclude_static=False, phase=2)
+    scene_mapping = model_folder.octree_scene_mapping(
+        exclude_static=False, phase=2, default_states=False)
     os.remove(r'./tests/assets/model_folders/simple/scene_mapping.json')
     assert 'two_phase' in scene_mapping
     assert 'three_phase' not in scene_mapping
     assert 'five_phase' not in scene_mapping
     assert len(scene_mapping['two_phase']) == 11
 
+def test_scene_mapping_3():
+    folder_path = r'./tests/assets/model_folders/simple/model'
+    model_folder = Folder.from_model_folder(folder_path)
+
     # three phase
-    scene_mapping = model_folder.octree_scene_mapping(exclude_static=False, phase=3)
+    scene_mapping = model_folder.octree_scene_mapping(
+        exclude_static=False, phase=3, default_states=False)
     os.remove(r'./tests/assets/model_folders/simple/scene_mapping.json')
     assert 'two_phase' in scene_mapping
     assert 'three_phase' in scene_mapping
     assert 'five_phase' not in scene_mapping
     assert len(scene_mapping['two_phase']) == 3
 
+def test_scene_mapping_5():
+    folder_path = r'./tests/assets/model_folders/simple/model'
+    model_folder = Folder.from_model_folder(folder_path)
+
     # five phase
-    scene_mapping = model_folder.octree_scene_mapping(exclude_static=False, phase=5)
+    scene_mapping = model_folder.octree_scene_mapping(
+        exclude_static=False, phase=5, default_states=False)
     os.remove(r'./tests/assets/model_folders/simple/scene_mapping.json')
     assert 'two_phase' in scene_mapping
     assert 'three_phase' in scene_mapping
@@ -39,15 +50,30 @@ def test_scene_mapping_static_model():
     model_folder = Folder.from_model_folder(folder_path)
 
     # exclude static apertures
-    scene_mapping = model_folder.octree_scene_mapping(exclude_static=False, phase=5)
+    scene_mapping = model_folder.octree_scene_mapping(
+        exclude_static=False, phase=5, default_states=False)
     os.remove(r'./tests/assets/model_folders/static/scene_mapping.json')
     assert len(scene_mapping['two_phase']) == 1
     assert len(scene_mapping['three_phase']) == 0
     assert len(scene_mapping['five_phase']) == 0
 
     # do not exclude static apertures
-    scene_mapping = model_folder.octree_scene_mapping(exclude_static=True, phase=5)
+    scene_mapping = model_folder.octree_scene_mapping(
+        exclude_static=True, phase=5, default_states=False)
     os.remove(r'./tests/assets/model_folders/static/scene_mapping.json')
     assert len(scene_mapping['two_phase']) == 0
     assert len(scene_mapping['three_phase']) == 0
     assert len(scene_mapping['five_phase']) == 0
+
+def test_scene_mapping_default_states():
+    folder_path = r'./tests/assets/model_folders/simple/model'
+    model_folder = Folder.from_model_folder(folder_path)
+
+    # two phase
+    scene_mapping = model_folder.octree_scene_mapping(
+        exclude_static=False, phase=2, default_states=True)
+    os.remove(r'./tests/assets/model_folders/simple/scene_mapping.json')
+    assert 'two_phase' in scene_mapping
+    assert 'three_phase' not in scene_mapping
+    assert 'five_phase' not in scene_mapping
+    assert len(scene_mapping['two_phase']) == 6