#2869 - Minor changes trying to make pytest happy

tests/conftest.py

@@ -7,9 +7,9 @@ from ray import init as rayinit
 
 from primaite import getLogger, PRIMAITE_PATHS
 from primaite.game.agent.actions import ActionManager
+from primaite.game.agent.interface import AbstractAgent
 from primaite.game.agent.observations.observation_manager import NestedObservation, ObservationManager
 from primaite.game.agent.rewards import RewardFunction
-from primaite.game.agent.interface import AbstractAgent
 from primaite.game.agent.scripted_agents.probabilistic_agent import ProbabilisticAgent
 from primaite.game.game import PrimaiteGame
 from primaite.simulator.file_system.file_system import FileSystem
@@ -369,6 +369,7 @@ def install_stuff_to_sim(sim: Simulation):
     # 5: Assert that the simulation starts off in the state that we expect
     assert len(sim.network.nodes) == 6
     assert len(sim.network.links) == 5
+
     # 5.1: Assert the router is correctly configured
     r = sim.network.router_nodes[0]
     for i, acl_rule in enumerate(r.acl.acl):

tests/integration_tests/configuration_file_parsing/software_installation_and_configuration.py

@@ -6,8 +6,8 @@ from typing import Union
 import yaml
 
 from primaite.config.load import data_manipulation_config_path
-from primaite.game.agent.scripted_agents.data_manipulation_bot import DataManipulationAgent
 from primaite.game.agent.interface import ProxyAgent
+from primaite.game.agent.scripted_agents.data_manipulation_bot import DataManipulationAgent
 from primaite.game.agent.scripted_agents.probabilistic_agent import ProbabilisticAgent
 from primaite.game.game import PrimaiteGame, SERVICE_TYPES_MAPPING
 from primaite.simulator.network.container import Network

tests/integration_tests/game_layer/observations/test_nic_observations.py

@@ -6,8 +6,8 @@ import pytest
 import yaml
 from gymnasium import spaces
 
-from primaite.game.agent.observations.nic_observations import NICObservation
 from primaite.game.agent.interface import ProxyAgent
+from primaite.game.agent.observations.nic_observations import NICObservation
 from primaite.game.game import PrimaiteGame
 from primaite.simulator.network.hardware.base import NetworkInterface
 from primaite.simulator.network.hardware.nodes.host.computer import Computer

tests/integration_tests/game_layer/test_rewards.py

@@ -2,8 +2,8 @@
 import pytest
 import yaml
 
-from primaite.game.agent.rewards import ActionPenalty, GreenAdminDatabaseUnreachablePenalty, WebpageUnavailablePenalty
 from primaite.game.agent.interface import AgentHistoryItem
+from primaite.game.agent.rewards import ActionPenalty, GreenAdminDatabaseUnreachablePenalty, WebpageUnavailablePenalty
 from primaite.game.game import PrimaiteGame
 from primaite.interface.request import RequestResponse
 from primaite.session.environment import PrimaiteGymEnv

tests/unit_tests/_primaite/_game/_agent/test_sticky_rewards.py

@@ -1,11 +1,11 @@
 # © Crown-owned copyright 2025, Defence Science and Technology Laboratory UK
 
+from primaite.game.agent.interface import AgentHistoryItem
 from primaite.game.agent.rewards import (
     GreenAdminDatabaseUnreachablePenalty,
     WebpageUnavailablePenalty,
     WebServer404Penalty,
 )
-from primaite.game.agent.interface import AgentHistoryItem
 from primaite.interface.request import RequestResponse