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})