tests/test_character_builder.py

@@ -3,8 +3,8 @@ import pathlib
 import unittest
 from unittest.mock import patch
 
-from elironwars.mob.base_mob import BaseMob
-from elironwars.mob.default_builder import CharacterBuilder
+from elironwars.mob import BaseMob, save_character, load_character
+from elironwars.mob.default_builder import build_character
 
 ARTHUR_JSON = """{
     "name": "Arthur",
@@ -37,7 +37,7 @@ class TestCharacterBuilder(unittest.TestCase):
         ]
 
         # Act
-        character = CharacterBuilder.build_character()
+        character = build_character()
 
         # Assert
         self.assertIsInstance(character, BaseMob)
@@ -60,7 +60,7 @@ class TestCharacterBuilder(unittest.TestCase):
         )
 
         # Act
-        CharacterBuilder.save_character(character)
+        save_character(character)
 
         # Assert
         self.assertTrue(ARTHUR_FILE.exists())
@@ -68,7 +68,7 @@ class TestCharacterBuilder(unittest.TestCase):
 
     def test_load_character_file_not_found(self):
         # Act
-        result = CharacterBuilder.load_character("Lancelot")
+        result = load_character("Lancelot")
 
         # Assert
         self.assertIsNone(result)
@@ -78,7 +78,7 @@ class TestCharacterBuilder(unittest.TestCase):
         ARTHUR_FILE.write_text(ARTHUR_JSON, encoding='utf-8')
 
         # Act
-        character = CharacterBuilder.load_character("Arthur")
+        character = load_character("Arthur")
 
         # Assert
         self.assertIsNotNone(character)
@@ -101,7 +101,7 @@ class TestCharacterBuilder(unittest.TestCase):
                     invalid_dict), encoding='utf-8')
 
                 # Act
-                result = CharacterBuilder.load_character("Arthur")
+                result = load_character("Arthur")
 
                 # Assert
                 self.assertIsNone(result)
@@ -111,7 +111,7 @@ class TestCharacterBuilder(unittest.TestCase):
         ARTHUR_FILE.write_text(ARTHUR_JSON[1:], encoding='utf-8')
 
         # Act
-        result = CharacterBuilder.load_character("Arthur")
+        result = load_character("Arthur")
 
         # Assert
         self.assertIsNone(result)