From d310543c0a448413f45a7750b13a047f42e96f7e Mon Sep 17 00:00:00 2001
From: vtexier <vit@free.fr>
Date: Thu, 19 Mar 2020 19:19:56 +0100
Subject: [PATCH] [fix] fix unit tests

---
 tests/unit/data/test_node_connector.py     | 2 +-
 tests/unit/data/test_nodes_repo.py         | 3 ++-
 tests/unit/money/test_quantitative_zsum.py | 4 ++--
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/tests/unit/data/test_node_connector.py b/tests/unit/data/test_node_connector.py
index cc9f465f..66ebcb63 100644
--- a/tests/unit/data/test_node_connector.py
+++ b/tests/unit/data/test_node_connector.py
@@ -1,4 +1,4 @@
-from duniterpy.documents import Peer
+from duniterpy.documents.peer import Peer
 from sakia.data.connectors import NodeConnector
 
 
diff --git a/tests/unit/data/test_nodes_repo.py b/tests/unit/data/test_nodes_repo.py
index 1ea58771..a440b382 100644
--- a/tests/unit/data/test_nodes_repo.py
+++ b/tests/unit/data/test_nodes_repo.py
@@ -1,6 +1,7 @@
 from sakia.data.repositories import NodesRepo
 from sakia.data.entities import Node
-from duniterpy.documents import BlockUID, BMAEndpoint, UnknownEndpoint, block_uid
+from duniterpy.documents import BlockUID, block_uid
+from duniterpy.api.endpoint import BMAEndpoint, UnknownEndpoint
 
 
 def test_add_get_drop_node(meta_repo):
diff --git a/tests/unit/money/test_quantitative_zsum.py b/tests/unit/money/test_quantitative_zsum.py
index f7269a29..d2531559 100644
--- a/tests/unit/money/test_quantitative_zsum.py
+++ b/tests/unit/money/test_quantitative_zsum.py
@@ -22,7 +22,7 @@ def test_localized_no_si(application_with_one_connection, bob):
         110, bob.currency, application_with_one_connection, None
     )
     value = referential.localized(units=True)
-    assert value == "-10.79 Q0 units"
+    assert value == "-10.79 Q0"
 
 
 def test_localized_with_si(application_with_one_connection, bob):
@@ -31,7 +31,7 @@ def test_localized_with_si(application_with_one_connection, bob):
         110 * 1000, bob.currency, application_with_one_connection, None
     )
     value = referential.localized(units=True, show_base=True)
-    assert value == "1,088.11 Q0 units"
+    assert value == "1,088.11 Q0"
 
 
 def test_localized_no_units_no_si(application_with_one_connection, bob):
-- 
GitLab