tests: Import from public modules when possible

Signed-off-by: Leandro Lucarella <[email protected]>

Author: llucax

tests/test_client.py

@@ -21,6 +21,7 @@
     Component,
     ComponentCategory,
     ComponentData,
+    Connection,
     EVChargerData,
     Fuse,
     GridMetadata,
@@ -29,7 +30,6 @@
     MeterData,
     MicrogridApiClient,
 )
-from frequenz.client.microgrid._connection import Connection
 
 
 class _TestClient(MicrogridApiClient):

tests/test_component.py

@@ -6,11 +6,11 @@
 import pytest
 from frequenz.api.common import components_pb2
 
-from frequenz.client.microgrid._component import (
+from frequenz.client.microgrid import (
     Component,
     ComponentCategory,
-    component_category_from_protobuf,
 )
+from frequenz.client.microgrid._component import component_category_from_protobuf
 
 
 def test_component_category_from_protobuf() -> None: