def test_transform_column_source_data_with_buffers(pd, cols, dt1, dt2):
d = dict(a=[1,2,3], b=np.array([4,5,6], dtype=dt1), c=pd.Series([7,8,9], dtype=dt2))
bufs = []
out = bus.transform_column_source_data(d, buffers=bufs, cols=cols)
assert set(out) == (set(d) if cols is None else set(cols))
if 'a' in out:
assert out['a'] == [1,2,3]
for x in ['b', 'c']:
dt = d[x].dtype
if x in out:
if dt in bus.BINARY_ARRAY_TYPES:
assert isinstance(out[x], dict)
assert 'shape' in out[x]
assert out[x]['shape'] == d[x].shape
assert 'dtype' in out[x]
assert out[x]['dtype'] == d[x].dtype.name
assert '__buffer__' in out[x]
else:
assert isinstance(out[x], list)
assert out[x] == list(d[x])
def test_set_data_from_json_nested_base64_and_list(self):
ds = ColumnDataSource()
data = {"foo": [np.arange(3), [1, 2, 3]]}
json = transform_column_source_data(data)
ds.set_from_json('data', json)
self.assertTrue(np.array_equal(ds.data["foo"], data["foo"]))
def test_set_data_from_json_base64(self):
ds = ColumnDataSource()
data = {"foo": np.arange(3)}
json = transform_column_source_data(data)
ds.set_from_json('data', json)
self.assertTrue(np.array_equal(ds.data["foo"], data["foo"]))
def test_set_data_from_json_nested_base64(self):
ds = ColumnDataSource()
data = {"foo": [[np.arange(3, dtype=np.int64)]]}
json = transform_column_source_data(data)
ds.set_from_json('data', json)
assert np.array_equal(ds.data["foo"], data["foo"])
def test_set_data_from_json_nested_base64(self):
ds = ColumnDataSource()
data = {"foo": [[np.arange(3)]]}
json = transform_column_source_data(data)
ds.set_from_json('data', json)
self.assertTrue(np.array_equal(ds.data["foo"], data["foo"]))
def test_set_data_from_json_nested_base64_and_list(self) -> None:
ds = bms.ColumnDataSource()
data = {"foo": [np.arange(3, dtype=np.int64), [1, 2, 3]]}
json = transform_column_source_data(data)
ds.set_from_json('data', json)
assert np.array_equal(ds.data["foo"], data["foo"])
def test_set_data_from_json_base64(self):
ds = bms.ColumnDataSource()
data = {"foo": np.arange(3, dtype=np.int64)}
json = transform_column_source_data(data)
ds.set_from_json('data', json)
assert np.array_equal(ds.data["foo"], data["foo"])
