import mock
import numpy as np
import xarray as xr
from ion_functions.data.ctd_functions import ctd_sbe16plus_tempwat, ctd_pracsal
from preload_database.database import initialize_connection, open_connection, PreloadDatabaseMode
from preload_database.model.preload import Parameter
from util.advlogging import jdefault
from util.calibration_coefficient_store import CalibrationCoefficientStore
from util.common import StreamKey, TimeRange
from util.stream_dataset import StreamDataset
TEST_DIR = os.path.dirname(__file__)
DATA_DIR = os.path.join(TEST_DIR, 'data')
initialize_connection(PreloadDatabaseMode.POPULATED_MEMORY)
open_connection()
logging.basicConfig()
log = logging.getLogger()
log.setLevel(logging.DEBUG)
class StreamDatasetTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.nutnr_sk = StreamKey('CE04OSPS', 'SF01B', '4A-NUTNRA102', 'streamed', 'nutnr_a_sample')
cls.ctdpf_sk = StreamKey('CE04OSPS', 'SF01B', '2A-CTDPFA107', 'streamed', 'ctdpf_sbe43_sample')
cls.nutnr_fn = 'nutnr_a_sample.nc'
cls.ctdpf_fn = 'ctdpf_sbe43_sample.nc'
cls.ctd_nutnr_cals = json.load(open(os.path.join(DATA_DIR, 'cals.json')))
import os
import unittest
import numpy as np
from datetime import datetime
import util.common as common
from preload_database.database import initialize_connection, open_connection, PreloadDatabaseMode
TEST_DIR = os.path.dirname(__file__)
initialize_connection(PreloadDatabaseMode.POPULATED_MEMORY)
open_connection()
class CommonTest(unittest.TestCase):
def setUp(self):
self.epoch_dt = datetime(1900, 1, 1, 0, 0, 0)
def test_isfillvalue(self):
assert np.array_equal(common.isfillvalue([1, 2, 3, -999999999, 4]),
[False, False, False, True, False])
assert np.array_equal(
common.isfillvalue([1.2, np.NaN, np.NaN, 2.3, np.NaN]),
[False, True, True, False, True])
assert np.array_equal(common.isfillvalue(['', 'abc']), [True, False])
def test_ntp_to_datetime_valid(self):
start_dt = datetime(2015, 1, 1, 12, 0, 0)
ntp_time = (start_dt - self.epoch_dt).total_seconds()
rval = common.ntp_to_datetime(ntp_time)
self.assertEqual(start_dt, rval)
def _initdb():
database.initialize_connection(database.PreloadDatabaseMode.POPULATED_MEMORY)
database.open_connection()