def test_init(self):
testname = "DataFlow.__init__()"
df = dataflow.DataFlow(self.inpds, self.outds)
actual = type(df)
expected = dataflow.DataFlow
print_test_message(testname, actual=actual, expected=expected)
self.assertIsInstance(df, expected, "{} failed".format(testname))
def test_execute_chunks_2D_x_y(self):
testname = 'DataFlow().execute()'
df = dataflow.DataFlow(self.inpds, self.outds)
expected = ValueError
print_test_message(testname, expected=expected)
self.assertRaises(expected, df.execute,
chunks=OrderedDict([('x', 4), ('y', 3)]))
def test_dimension_map(self):
testname = 'DataFlow().dimension_map'
df = dataflow.DataFlow(self.inpds, self.outds)
actual = df.dimension_map
expected = {'lat': 'y', 'strlen': 'n', 'lon': 'x',
'ncat': 'c', 'time': 't', 'bnds': 'd'}
print_test_message(testname, actual=actual, expected=expected)
self.assertEqual(actual, expected, '{} failed'.format(testname))
def test_execute_chunks_2D_t_y(self):
testname = "DataFlow().execute()"
df = dataflow.DataFlow(self.inpds, self.outds)
df.execute(chunks=OrderedDict([("t", 2), ("y", 3)]))
actual = all(exists(f) for f in self.outfiles.values())
expected = True
print_test_message(testname, actual=actual, expected=expected)
self.assertEqual(actual, expected, "{} failed".format(testname))
for f in self.outfiles:
print_ncfile(self.outfiles[f])
print
def test_execute_all(self):
testname = "DataFlow().execute()"
df = dataflow.DataFlow(self.inpds, self.outds)
df.execute(history=True)
actual = all(exists(f) for f in self.outfiles.values())
expected = True
print_test_message(testname, actual=actual, expected=expected)
self.assertEqual(actual, expected, "{} failed".format(testname))
for f in self.outfiles:
print_ncfile(self.outfiles[f])
print
def test_execute_chunks_1D_y(self):
testname = 'DataFlow().execute()'
df = dataflow.DataFlow(self.inpds, self.outds)
df.execute(chunks={'y': 3})
actual = all(exists(f) for f in self.outfiles.itervalues())
expected = True
print_test_message(testname, actual=actual, expected=expected)
self.assertEqual(actual, expected, '{} failed'.format(testname))
for f in self.outfiles:
print_ncfile(self.outfiles[f])
print
def test_dimension_map(self):
testname = "DataFlow().dimension_map"
df = dataflow.DataFlow(self.inpds, self.outds)
actual = df.dimension_map
expected = {
"lat": "y",
"strlen": "n",
"lon": "x",
"ncat": "c",
"time": "t",
"bnds": "d",
}
print_test_message(testname, actual=actual, expected=expected)
self.assertEqual(actual, expected, "{} failed".format(testname))
def test_execute_chunks_1D_x(self):
testname = "DataFlow().execute()"
df = dataflow.DataFlow(self.inpds, self.outds)
expected = ValueError
print_test_message(testname, expected=expected)
self.assertRaises(expected, df.execute, chunks={"x": 4})
