def test_surface_and_times(self, to_datetime): valid = datetime(2019, 10, 10, 9) initial = datetime(2019, 10, 10, 3) to_datetime.side_effect = [valid, initial] result = ghrsstl4.coordinates(sentinel.valid, sentinel.initial, [], None) self.assertEqual(to_datetime.mock_calls, [call(sentinel.valid), call(sentinel.initial)]) self.assertEqual(result, {'valid': [valid], 'initial': [initial], 'length': ['T+6'], 'level': ['Sea Surface']})
def test_surface_no_pressure(self, to_datetime): result = ghrsstl4.coordinates(None, None, [1000, 900], None) self.assertEqual(result['level'], ['Sea Surface'])