Skip to content

Commit 139a731

Browse files
committed
merge master into fill_values
2 parents dbb7c6d + d8c8838 commit 139a731

2 files changed

Lines changed: 7 additions & 7 deletions

File tree

ncwriter/template.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
import netCDF4
2121

22-
from schema import validate_dimensions, validate_variables, validate_attributes
22+
from .schema import validate_dimensions, validate_variables, validate_attributes
2323

2424

2525
class NetCDFGroupDict(object):
@@ -328,7 +328,7 @@ def update_dimensions(self):
328328
Otherwise raise ValueError. Also raise ValueError if a dimension that already has a non-zero size is not
329329
consistent with variable array sizes.
330330
"""
331-
for name, var in self.variables.iteritems():
331+
for name, var in self.variables.items():
332332
values = var.get('data')
333333
if values is None:
334334
continue
@@ -368,7 +368,7 @@ def createVariables(self, **kwargs):
368368
**kwargs are included here to overload all options for all variables
369369
like `zlib` and friends.
370370
"""
371-
for varname, var in self.variables.iteritems():
371+
for varname, var in self.variables.items():
372372
datatype = var['type']
373373
dimensions = var['dimensions']
374374
cwargs = kwargs.copy()

test_ncwriter/test_template.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -105,12 +105,12 @@ def test_update_dimensions(self):
105105
template = DatasetTemplate.from_json(TEMPLATE_JSON)
106106
template.dimensions['TIME'] = 100
107107
template.dimensions['DEPTH'] = 10
108-
self.assertDictContainsSubset(OrderedDict([('TIME', 100), ('DEPTH', 10)]), template.dimensions)
108+
self.assertEqual(OrderedDict([('TIME', 100), ('DEPTH', 10)]), template.dimensions)
109109

110110
def test_add_variables(self):
111111
template = DatasetTemplate.from_json(TEMPLATE_PARTIAL_JSON)
112112
template.variables['TIME'] = self.variables['TIME']
113-
self.assertEqual(['TEMP', 'TIME'], template.variables.keys())
113+
self.assertEqual({'TEMP', 'TIME'}, set(template.variables.keys()))
114114
self.assertEqual(self.variables['TIME'], template.variables['TIME'])
115115

116116
def test_add_variable_dimensions(self):
@@ -168,10 +168,10 @@ def test_create_file(self):
168168
('TIME', len(self.values10)),
169169
('DEPTH', len(self.values1))
170170
])
171-
ds_dimensions = OrderedDict((k, v.size) for k, v in dataset.dimensions.iteritems())
171+
ds_dimensions = OrderedDict((k, v.size) for k, v in dataset.dimensions.items())
172172
self.assertEqual(expected_dimensions, ds_dimensions)
173173

174-
for vname, vdict in self.variables.iteritems():
174+
for vname, vdict in self.variables.items():
175175
ds_var = dataset[vname]
176176
self.assertEqual(vdict['dimensions'], list(ds_var.dimensions))
177177
self.assertEqual(vdict['type'], ds_var.dtype)

0 commit comments

Comments
 (0)