factor out counting dict

Author: alimanfoo

zarr/tests/test_attrs.py

@@ -2,7 +2,6 @@
 from __future__ import absolute_import, print_function, division
 import json
 import unittest
-import collections
 
 
 from nose.tools import eq_ as eq, assert_raises
@@ -11,34 +10,7 @@
 from zarr.attrs import Attributes
 from zarr.compat import binary_type, text_type
 from zarr.errors import PermissionError
-
-
-class CountingDict(collections.MutableMapping):
-
-    def __init__(self):
-        self.wrapped = dict()
-        self.counter = collections.Counter()
-
-    def __len__(self):
-        return len(self.wrapped)
-
-    def __iter__(self):
-        return iter(self.wrapped)
-
-    def __contains__(self, item):
-        return item in self.wrapped
-
-    def __getitem__(self, item):
-        self.counter['__getitem__', item] += 1
-        return self.wrapped[item]
-
-    def __setitem__(self, key, value):
-        self.counter['__setitem__', key] += 1
-        self.wrapped[key] = value
-
-    def __delitem__(self, key):
-        self.counter['__delitem__', key] += 1
-        del self.wrapped[key]
+from zarr.tests.util import CountingDict
 
 
 class TestAttributes(unittest.TestCase):

zarr/tests/util.py

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+from __future__ import absolute_import, print_function, division
+import collections
+
+
+class CountingDict(collections.MutableMapping):
+
+    def __init__(self):
+        self.wrapped = dict()
+        self.counter = collections.Counter()
+
+    def __len__(self):
+        return len(self.wrapped)
+
+    def __iter__(self):
+        return iter(self.wrapped)
+
+    def __contains__(self, item):
+        return item in self.wrapped
+
+    def __getitem__(self, item):
+        self.counter['__getitem__', item] += 1
+        return self.wrapped[item]
+
+    def __setitem__(self, key, value):
+        self.counter['__setitem__', key] += 1
+        self.wrapped[key] = value
+
+    def __delitem__(self, key):
+        self.counter['__delitem__', key] += 1
+        del self.wrapped[key]