Add DataDict in Range

Author: seignovert

opus/range.py

@@ -1,11 +1,15 @@
 # -*- coding: utf-8 -*-
 
-class Range(object):
+from .data import DataDict
+
+class Range(DataDict):
     def __init__(self, field, json):
-        self.max = float(json['max'])
-        self.min = float(json['min'])
-        self.nulls = int(json['nulls'])
+        DataDict.__init__(self)
         self._field = field
+        self.append('min', float(json['min']))
+        self.append('max', float(json['max']))
+        self.append('null', int(json['nulls']))
 
     def __repr__(self):
-        return 'OPUS API Range endpoints for field: {}'.format(self._field)
+        return 'OPUS API Range endpoints for field: `{}`\n'.format(self._field) + \
+               '\n'.join(' - {} : {}'.format(key, value) for key, value in self.items())

tests/test_api.py

@@ -275,10 +275,10 @@ def test_api_range(api):
     assert responses.calls[0].request.url == 'http://localhost/meta/range/endpoints/ringradius1.json?target=Saturn'
     assert responses.calls[0].response.text == range_endpoints
 
-    assert repr(resp) == 'OPUS API Range endpoints for field: ringradius1'
-    assert resp.min == 60000
-    assert resp.max == 1.29e+07
-    assert resp.nulls == 115573
+    assert 'OPUS API Range endpoints for field: `ringradius1`' in repr(resp)
+    assert resp['min'] == 60000
+    assert resp['max'] == 1.29e+07
+    assert resp['null'] == 115573
 
 
 @responses.activate