def test_numpy_timelike_column_with_null(dsn, configuration):
fill_value = 0;
with open_cursor(configuration) as cursor:
with query_fixture(cursor, configuration, 'INSERT TIMESTAMP') as table_name:
cursor.execute('INSERT INTO {} VALUES (?)'.format(table_name), [None])
cursor.execute('SELECT a FROM {}'.format(table_name))
results = cursor.fetchallnumpy()
expected = MaskedArray([42], mask=[1], dtype='datetime64[us]')
assert_equal(results[_fix_case(configuration, 'a')].filled(fill_value),
expected.filled(fill_value))
def test_numpy_timelike_column_with_null(dsn, configuration):
fill_value = 0;
with open_cursor(configuration) as cursor:
with query_fixture(cursor, configuration, 'INSERT TIMESTAMP') as table_name:
cursor.execute('INSERT INTO {} VALUES (?)'.format(table_name), [None])
cursor.execute('SELECT a FROM {}'.format(table_name))
results = cursor.fetchallnumpy()
expected = MaskedArray([42], mask=[1], dtype='datetime64[us]')
assert_equal(results[_fix_case(configuration, 'a')].filled(fill_value),
expected.filled(fill_value))
def test_numpy_timelike_column_with_null(dsn, configuration):
fill_value = 0
with open_cursor(configuration) as cursor:
with query_fixture(cursor, configuration,
"INSERT TIMESTAMP") as table_name:
cursor.execute(f"INSERT INTO {table_name} VALUES (?)", [None])
cursor.execute(f"SELECT a FROM {table_name}")
results = cursor.fetchallnumpy()
expected = MaskedArray([42], mask=[1], dtype="datetime64[us]")
assert_equal(
results[_fix_case(configuration, "a")].filled(fill_value),
expected.filled(fill_value),
)
def _plot_bar_data(ax, data: ma.MaskedArray, time: ndarray):
"""Plots 1D variable as bar plot.
Args:
ax (obj): Axes object.
data (maskedArray): 1D data array.
time (ndarray): 1D time array.
"""
# TODO: unit change somewhere else
ax.plot(time, data / 1000, color="navy")
ax.bar(time,
data.filled(0) / 1000,
width=1 / 120,
align="center",
alpha=0.5,
color="royalblue")
pos = ax.get_position()
ax.set_position([pos.x0, pos.y0, pos.width * 0.965, pos.height])
