checkpoint at 2024-06-21T17:13:11+0800

miyuki-shirogane · Jun 21, 2024 · 526f58a · 526f58a
1 parent 745c00e
commit 526f58a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 34 deletions.
diff --git a/pistol_magazine/self_made.py b/pistol_magazine/self_made.py
@@ -102,37 +102,6 @@ def model_to_data(cls, data: dict or list) -> Dict or List:
     def get_datatype(self):
         return self.models.get_datatype()
 
-    # def mock(
-    #         self, to_json: bool = False,
-    #         num_entries: Optional[int] = None,
-    #         key_generator: Optional[Callable[[], str]] = None,
-    #         as_list: bool = False
-    # ):
-    #     if key_generator is None:
-    #         key_generator = lambda: str(uuid.uuid4())
-    #     hook_manager.trigger_hooks('pre_generate', None)
-    #     if num_entries is not None:
-    #         if as_list:
-    #             final_result = []
-    #             for _ in range(num_entries):
-    #                 data = self.models.mock(to_json=False)
-    #                 data = hook_manager.trigger_hooks('after_generate', data)
-    #                 final_result.append(data)
-    #         else:
-    #             final_result = {}
-    #             for _ in range(num_entries):
-    #                 entry_key = key_generator()
-    #                 data = self.models.mock(to_json=False)
-    #                 data = hook_manager.trigger_hooks('after_generate', data)
-    #                 final_result[entry_key] = data
-    #         result = final_result
-    #     else:
-    #         result = self.models.mock(to_json=False)
-    #         result = hook_manager.trigger_hooks('after_generate', result)
-    #     if to_json:
-    #         result = json.dumps(result)
-    #     result = hook_manager.trigger_hooks('final_generate', result)
-    #     return result
     def mock(
             self, to_json: bool = False,
             num_entries: Optional[int] = None,

diff --git a/tests/test_selfmade.py b/tests/test_selfmade.py
@@ -1,8 +1,7 @@
 from pprint import pprint
-from random import choice
-from pistol_magazine import DataMocker, provider, Str, Int, Timestamp, Bool, Dict, Float, List, Datetime, StrInt
+from random import choice, randint
+from pistol_magazine import *
 from pistol_magazine.hooks.hooks import hook
-from pistol_magazine.self_made import ProviderField
 
 
 def test_model_data_conversion():