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"])