Skip to content

Commit

Permalink
dev(hansbug): move CustomMapping for test.testings
Browse files Browse the repository at this point in the history
  • Loading branch information
HansBug committed Oct 22, 2023
1 parent a323641 commit ab350d6
Show file tree
Hide file tree
Showing 4 changed files with 18 additions and 28 deletions.
1 change: 1 addition & 0 deletions test/testings/__init__.py
@@ -0,0 +1 @@
from .mapping import CustomMapping
15 changes: 15 additions & 0 deletions test/testings/mapping.py
@@ -0,0 +1,15 @@
import collections.abc


class CustomMapping(collections.abc.Mapping):
def __init__(self, **kwargs):
self._kwargs = kwargs

def __getitem__(self, __key):
return self._kwargs[__key]

def __len__(self):
return len(self._kwargs)

def __iter__(self):
yield from self._kwargs
14 changes: 1 addition & 13 deletions test/tree/general/base.py
Expand Up @@ -11,21 +11,9 @@
from treevalue import register_dict_type
from treevalue.tree import func_treelize, TreeValue, raw, mapping, delayed, FastTreeValue
from ..tree.base import get_treevalue_test
from ...testings import CustomMapping


class CustomMapping(collections.abc.Mapping):
def __init__(self, **kwargs):
self._kwargs = kwargs

def __getitem__(self, __key):
return self._kwargs[__key]

def __len__(self):
return len(self._kwargs)

def __iter__(self):
yield from self._kwargs


def get_fasttreevalue_test(treevalue_class: Type[FastTreeValue]):
class Container:
Expand Down
16 changes: 1 addition & 15 deletions test/tree/tree/base.py
@@ -1,4 +1,3 @@
import collections.abc
import pickle
import re
import unittest
Expand All @@ -11,6 +10,7 @@
from treevalue import raw, TreeValue, delayed, ValidationError, register_dict_type
from treevalue.tree.common import create_storage
from treevalue.tree.tree.constraint import cleaf
from ...testings import CustomMapping

try:
_ = reversed({}.keys())
Expand Down Expand Up @@ -44,20 +44,6 @@ def __hash__(self):
return hash((self.__value,))


class CustomMapping(collections.abc.Mapping):
def __init__(self, **kwargs):
self._kwargs = kwargs

def __getitem__(self, __key):
return self._kwargs[__key]

def __len__(self):
return len(self._kwargs)

def __iter__(self):
yield from self._kwargs


def get_treevalue_test(treevalue_class: Type[TreeValue]):
# noinspection DuplicatedCode,PyMethodMayBeStatic

Expand Down

0 comments on commit ab350d6

Please sign in to comment.