Merge branch 'cleanup' of github.com:bit-bots/humanoid_league_misc in…

…to cleanup
bit-bots · Nov 12, 2023 · 872ed6b · 872ed6b
2 parents cb481a9 + e0eb742
commit 872ed6b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/..._communication/humanoid_league_team_communication/converter/robocup_protocol_converter.py b/..._communication/humanoid_league_team_communication/converter/robocup_protocol_converter.py
@@ -46,8 +46,8 @@ def __init__(self, own_team_color: TeamColor):
 
         self.proto_to_team_data_team_mapping = {
             Proto.Team.UNKNOWN_TEAM: RobotRelative.ROBOT_UNDEFINED,
-            Proto.Team.BLUE: RobotRelative.ROBOT_CYAN,
-            Proto.Team.RED: RobotRelative.ROBOT_MAGENTA
+            Proto.Team.BLUE: RobotRelative.ROBOT_BLUE,
+            Proto.Team.RED: RobotRelative.ROBOT_RED
         }
         self.state_to_proto_team_mapping = {
             RobotAttributes.TEAM_OWN: Proto.Team.RED if own_team_color == TeamColor.RED else Proto.Team.BLUE,

diff --git a/humanoid_league_team_communication/test/converter/test_message_to_team_data_converter.py b/humanoid_league_team_communication/test/converter/test_message_to_team_data_converter.py
@@ -32,8 +32,8 @@ def test_convert_robots(snapshot, message_with_other_robots, team_data_with_head
     team_data = convert_from_message(message_with_other_robots, team_data_with_header)
 
     assert team_data.robots.header == team_data_with_header.header
-    assert team_data.robots.robots[0].type == RobotRelative.ROBOT_CYAN
-    assert team_data.robots.robots[1].type == RobotRelative.ROBOT_MAGENTA
+    assert team_data.robots.robots[0].type == RobotRelative.ROBOT_BLUE
+    assert team_data.robots.robots[1].type == RobotRelative.ROBOT_RED
 
     assert str(team_data.robots) == snapshot