diff options
author | Petr Viktorin <pviktori@redhat.com> | 2015-08-11 13:51:14 +0200 |
---|---|---|
committer | Jan Cholasta <jcholast@redhat.com> | 2015-09-01 11:42:01 +0200 |
commit | 3bf91eab25c602a6fad2665456f57e8629c5a6f4 (patch) | |
tree | 52f713e898a385d57a914d539a7da9a20fc20166 /ipatests/test_ipapython/test_ipautil.py | |
parent | dd16cc98b0d67f1448bf9de25f8adce512b1431c (diff) | |
download | freeipa-3bf91eab25c602a6fad2665456f57e8629c5a6f4.tar.gz freeipa-3bf91eab25c602a6fad2665456f57e8629c5a6f4.tar.xz freeipa-3bf91eab25c602a6fad2665456f57e8629c5a6f4.zip |
Use Python3-compatible dict method names
Python 2 has keys()/values()/items(), which return lists,
iterkeys()/itervalues()/iteritems(), which return iterators,
and viewkeys()/viewvalues()/viewitems() which return views.
Python 3 has only keys()/values()/items(), which return views.
To get iterators, one can use iter() or a for loop/comprehension;
for lists there's the list() constructor.
When iterating through the entire dict, without modifying the dict,
the difference between Python 2's items() and iteritems() is
negligible, especially on small dicts (the main overhead is
extra memory, not CPU time). In the interest of simpler code,
this patch changes many instances of iteritems() to items(),
iterkeys() to keys() etc.
In other cases, helpers like six.itervalues are used.
Reviewed-By: Christian Heimes <cheimes@redhat.com>
Reviewed-By: Jan Cholasta <jcholast@redhat.com>
Diffstat (limited to 'ipatests/test_ipapython/test_ipautil.py')
-rw-r--r-- | ipatests/test_ipapython/test_ipautil.py | 29 |
1 files changed, 17 insertions, 12 deletions
diff --git a/ipatests/test_ipapython/test_ipautil.py b/ipatests/test_ipapython/test_ipautil.py index 112759b01..ee0f559cf 100644 --- a/ipatests/test_ipapython/test_ipautil.py +++ b/ipatests/test_ipapython/test_ipautil.py @@ -24,6 +24,7 @@ import sys import nose import pytest +import six from ipapython import ipautil @@ -132,9 +133,13 @@ class TestCIDict(object): self.cidict.clear() nose.tools.assert_equal(0, len(self.cidict)) assert self.cidict == {} - assert self.cidict.keys() == [] - assert self.cidict.values() == [] - assert self.cidict.items() == [] + assert list(self.cidict) == [] + assert list(self.cidict.values()) == [] + assert list(self.cidict.items()) == [] + if six.PY2: + assert self.cidict.keys() == [] + assert self.cidict.values() == [] + assert self.cidict.items() == [] assert self.cidict._keys == {} def test_copy(self): @@ -145,7 +150,7 @@ class TestCIDict(object): assert "key1" in copy nose.tools.assert_equal("val1", copy["Key1"]) - @pytest.mark.skipif(sys.version_info >= (3, 0), reason="Python 2 only") + @pytest.mark.skipif(not six.PY2, reason="Python 2 only") def test_haskey(self): assert self.cidict.has_key("KEY1") assert self.cidict.has_key("key2") @@ -161,15 +166,15 @@ class TestCIDict(object): assert "Key4" not in self.cidict def test_items(self): - items = self.cidict.items() + items = list(self.cidict.items()) nose.tools.assert_equal(3, len(items)) items_set = set(items) assert ("Key1", "val1") in items_set assert ("key2", "val2") in items_set assert ("KEY3", "VAL3") in items_set - assert self.cidict.items() == list(self.cidict.iteritems()) == zip( - self.cidict.iterkeys(), self.cidict.itervalues()) + assert list(self.cidict.items()) == list(self.cidict.iteritems()) == zip( + self.cidict.keys(), self.cidict.values()) def test_iter(self): items = [] @@ -207,24 +212,24 @@ class TestCIDict(object): assert "VAL3" in values_set def test_keys(self): - keys = self.cidict.keys() + keys = list(self.cidict.keys()) nose.tools.assert_equal(3, len(keys)) keys_set = set(keys) assert "Key1" in keys_set assert "key2" in keys_set assert "KEY3" in keys_set - assert self.cidict.keys() == list(self.cidict.iterkeys()) + assert list(self.cidict.keys()) == list(self.cidict.iterkeys()) def test_values(self): - values = self.cidict.values() + values = list(self.cidict.values()) nose.tools.assert_equal(3, len(values)) values_set = set(values) assert "val1" in values_set assert "val2" in values_set assert "VAL3" in values_set - assert self.cidict.values() == list(self.cidict.itervalues()) + assert list(self.cidict.values()) == list(self.cidict.itervalues()) def test_update(self): newdict = { "KEY2": "newval2", @@ -232,7 +237,7 @@ class TestCIDict(object): self.cidict.update(newdict) nose.tools.assert_equal(4, len(self.cidict)) - items = self.cidict.items() + items = list(self.cidict.items()) nose.tools.assert_equal(4, len(items)) items_set = set(items) assert ("Key1", "val1") in items_set |