update save name for these

tests/mock_robot/test_mock_env_cup_emptiness.py

@@ -102,7 +102,7 @@ def test_with_belief_observe_cup_emptiness():
     }
 
     # Plan and visualize transitions
-    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name="transition_graph")
+    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=f'Transition Graph, {test_name.replace("_", " ").title()}')
 
     # Verify transition graph file exists
     graph_file = Path(test_dir) / "transitions" / "transition_graph.png"
@@ -285,10 +285,11 @@ def test_with_belief_plan_check_and_pick_cup():
     }
 
     # Plan and visualize transitions
-    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name="transition_graph")
+    name = f'Transition Graph, {test_name.replace("_", " ").title()}'
+    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=name)
 
     # Verify transition graph file exists
-    graph_file = Path(test_dir) / "transitions" / "transition_graph.png"
+    graph_file = Path(test_dir) / "transitions" / f"{name}.png"
     assert graph_file.exists(), "Transition graph file not generated"
 
     # Cleanup
@@ -384,7 +385,7 @@ def test_with_belief_plan_check_and_place_cup():
     }
 
     # Plan and visualize transitions
-    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name="transition_graph")
+    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=f'Transition Graph, {test_name.replace("_", " ").title()}')
 
     # Verify transition graph file exists
     graph_file = Path(test_dir) / "transitions" / "transition_graph.png"
@@ -472,7 +473,7 @@ def test_view_pick_and_place_cup():
     }
 
     # Plan and visualize transitions
-    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name="transition_graph")
+    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=f'Transition Graph, {test_name.replace("_", " ").title()}')
 
     # Verify transition graph file exists
     graph_file = Path(test_dir) / "transitions" / "transition_graph.png"
@@ -592,7 +593,7 @@ def test_observe_two_cups_and_place_empty():
     }
 
     # Plan and visualize transitions
-    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name="transition_graph")
+    env_creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=f'Transition Graph, {test_name.replace("_", " ").title()}')
 
     # Verify transition graph file exists
     graph_file = Path(test_dir) / "transitions" / "transition_graph.png"

tests/mock_robot/test_mock_env_graph_building.py

@@ -168,7 +168,8 @@ def test_transitions_match_edges():
     }
 
     # Plan and visualize first to generate the graph
-    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=test_name)
+    name = f'Transition Graph, {test_name.replace("_", " ").title()}'
+    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=name)
 
     # Get all possible operator transitions
     transitions = creator.get_operator_transitions(initial_atoms, objects)

tests/mock_robot/test_mock_env_transitions.py

@@ -56,6 +56,7 @@
 from rich.console import Console
 from rich.logging import RichHandler
 import logging
+from pathlib import Path
 
 # Configure rich logging
 logging.basicConfig(
@@ -166,10 +167,12 @@ def test_single_block_pick_place():
     }
 
     # Plan and visualize transitions
-    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name="transition_graph")
+    name = f'Transition Graph, {test_name.replace("_", " ").title()}'
+    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=name)
 
     # Verify output file exists
-    assert os.path.exists(os.path.join(creator.transitions_dir, "transition_graph.png"))
+    graph_file = Path(os.path.join("mock_env_data", test_name)) / "transitions" / f"{name}.png"
+    assert graph_file.exists(), "Transition graph file not generated"
 
 
 def test_two_object_pick_place():
@@ -266,10 +269,12 @@ def test_two_object_pick_place():
     }
 
     # Plan and visualize transitions
-    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name="transition_graph")
+    name = f'Transition Graph, {test_name.replace("_", " ").title()}'
+    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=name)
 
     # Verify output file exists
-    assert os.path.exists(os.path.join(creator.transitions_dir, "transition_graph.png"))
+    graph_file = Path(os.path.join("mock_env_data", test_name)) / "transitions" / f"{name}.png"
+    assert graph_file.exists(), "Transition graph file not generated"
 
 
 def test_view_reach_pick_place_two_objects():
@@ -367,7 +372,9 @@ def test_view_reach_pick_place_two_objects():
     }
 
     # Plan and visualize transitions
-    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name="transition_graph")
+    name = f'Transition Graph, {test_name.replace("_", " ").title()}'
+    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=name)
 
     # Verify output file exists
-    assert os.path.exists(os.path.join(creator.transitions_dir, "transition_graph.png")) 
+    graph_file = Path(os.path.join("mock_env_data", test_name)) / "transitions" / f"{name}.png"
+    assert graph_file.exists(), "Transition graph file not generated" 

tests/mock_robot/test_mock_env_view.py

@@ -50,7 +50,7 @@ def test_view_operators():
     }
 
     # Plan and visualize transitions
-    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name="transition_graph")
+    creator.plan_and_visualize(initial_atoms, goal_atoms, objects, task_name=f'Transition Graph, {test_name.replace("_", " ").title()}')
 
     # Verify output file exists
     assert os.path.exists(os.path.join(creator.transitions_dir, "transition_graph.png"))