diff options
| -rw-r--r-- | netlib/odict.py | 6 | ||||
| -rw-r--r-- | test/test_odict.py | 6 |
2 files changed, 12 insertions, 0 deletions
diff --git a/netlib/odict.py b/netlib/odict.py index afc33caa..629fcade 100644 --- a/netlib/odict.py +++ b/netlib/odict.py @@ -80,6 +80,12 @@ class ODict: else: return d + def get_first(self, k, d=None): + if k in self: + return self[k][0] + else: + return d + def items(self): return self.lst[:] diff --git a/test/test_odict.py b/test/test_odict.py index e7453e2d..f27f6f8b 100644 --- a/test/test_odict.py +++ b/test/test_odict.py @@ -85,6 +85,12 @@ class TestODict: assert self.od.get("one") == ["two"] assert self.od.get("two") == None + def test_get_first(self): + self.od.add("one", "two") + self.od.add("one", "three") + assert self.od.get_first("one") == "two" + assert self.od.get_first("two") == None + class TestODictCaseless: def setUp(self): |