mergeabledict 테스트 추가

laoraid · May 21, 2020 · af7e311 · af7e311
1 parent 49918bf
commit af7e311
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/KartRider/basedata.py b/KartRider/basedata.py
@@ -50,8 +50,11 @@ def mergeValues(self) -> list:
         length = sum(len(x) for x in self.values())
         li = [None] * length
 
+        i = 0
+
         for v in self.values():
-            for i, rr in enumerate(v):
+            for rr in v:
                 li[i] = rr
+                i += 1
 
         return li
diff --git a/tests/test_etc.py b/tests/test_etc.py
@@ -46,3 +46,12 @@ def test_isid():
     for k, v in zip(keys, values):
         assert utils._isId(k)
         assert not utils._isId(v)
+
+
+def test_mergedict():
+    from KartRider.basedata import MergeAbleDict
+
+    mdict = MergeAbleDict(abc=['a', 'b', 'c'], wegh=['d', 'e'],
+                          wegweg=['f', 'g', 'h', 'i'])
+
+    assert ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i'] == mdict.mergeValues()