Merge pull request #118 from dschep/master

Raise an error if selecting column that exists more than once
This commit is contained in:
2017-12-11 07:14:50 -05:00
committed by GitHub
2 changed files with 8 additions and 0 deletions
+2
View File
@@ -52,6 +52,8 @@ class Record(object):
# Support for string-based lookup.
if key in self.keys():
i = self.keys().index(key)
if self.keys().count(key) > 1:
raise KeyError("Record contains multiple '{}' fields.".format(key))
return self.values()[i]
raise KeyError("Record contains no '{}' field.".format(key))
+6
View File
@@ -97,3 +97,9 @@ class TestRecord:
assert key in _dir
for key in dir(object):
assert key in _dir
def test_record_duplicate_column(self):
keys, values = ['id', 'name', 'email', 'email'], [1, '', '', '']
record = records.Record(keys, values)
with raises(KeyError):
record['email']