def test_viewer_load_named_table(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer()
viewer.load(table, name="data_1")
assert viewer.columns == ["a"]
assert viewer.table_name == "data_1"
assert viewer.table == table
def test_viewer_load_table_with_options(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer()
# options should be disregarded when loading Table
viewer.load(table, limit=1)
assert viewer.columns == ["a"]
assert viewer.table.size() == 3
def test_viewer_load_df(self):
table = Table(pd.DataFrame({"a": np.arange(1, 100)}))
viewer = PerspectiveViewer()
viewer.load(table)
for col in viewer.columns:
assert col in ["index", "a"]
assert viewer.table.size() == 99
def test_viewer_load_view(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer()
view = table.view()
viewer.load(view)
assert viewer._view == view
assert viewer.table == table
def test_viewer_load_schema(self):
viewer = PerspectiveViewer()
viewer.load({
"a": str,
"b": int,
"c": bool,
"d": str
})
for col in viewer.columns:
assert col in ["a", "b", "c", "d"]
def test_save_restore(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer(plugin="x_bar",
filters=[["a", "==", 2]],
editable=True,
expressions=['"a" * 2'])
viewer.load(table)
# Save config
config = viewer.save()
assert viewer.filters == [["a", "==", 2]]
assert config["filters"] == [["a", "==", 2]]
assert viewer.plugin == "x_bar"
assert config["plugin"] == "x_bar"
assert config["editable"] is True
assert config["expressions"] == ['"a" * 2']
# reset configuration
viewer.reset()
assert viewer.plugin == "datagrid"
assert viewer.filters == []
assert viewer.editable is False
assert viewer.expressions == []
# restore configuration
viewer.restore(**config)
assert viewer.filters == [["a", "==", 2]]
assert viewer.plugin == "x_bar"
assert viewer.editable is True
assert viewer.expressions == ['"a" * 2']
def test_viewer_make_view_replace(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer(filters=[["a", "==", 2]])
viewer.load(table)
viewer._new_view()
assert viewer.view.to_dict() == {"a": [2]}
viewer.filters = []
viewer._new_view()
assert viewer.view.to_dict() == {"a": [1, 2, 3]}
def test_viewer_clear(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer()
viewer.load(table)
viewer.clear()
assert viewer.table.size() == 0
assert viewer.table.schema() == {"a": int}
def test_viewer_delete(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer(plugin="x_bar", filters=[["a", "==", 2]])
viewer.load(table)
assert viewer.filters == [["a", "==", 2]]
viewer.delete()
assert viewer.table_name is None
assert viewer.table is None
def test_viewer_load_clears_state(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer(theme="Material Dark", group_by=["a"])
viewer.load(table)
assert viewer.group_by == ["a"]
viewer.load({"b": [1, 2, 3]})
assert viewer.group_by == []
assert viewer.theme == "Material Dark" # should not break UI
def test_viewer_load_clears_state(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer(dark=True, row_pivots=["a"])
viewer.load(table)
assert viewer.row_pivots == ["a"]
viewer.load({"b": [1, 2, 3]})
assert viewer.row_pivots == []
assert viewer.dark is True # should not break UI
def test_viewer_reset(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer(plugin="x_bar", filters=[["a", "==", 2]])
viewer.load(table)
assert viewer.filters == [["a", "==", 2]]
viewer.reset()
assert viewer.plugin == "hypergrid"
assert viewer.filters == []
def test_viewer_update_list(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer()
viewer.load(table)
viewer.update([{"a": 4}, {"a": 5}, {"a": 6}])
assert table.size() == 6
assert viewer.table.size() == 6
assert viewer.table.view().to_dict() == {"a": [1, 2, 3, 4, 5, 6]}
def test_viewer_update_df(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer()
viewer.load(table)
viewer.update(pd.DataFrame({"a": [4, 5, 6]}))
assert table.size() == 6
assert viewer.table.size() == 6
assert viewer.table.view().to_dict() == {"a": [1, 2, 3, 4, 5, 6]}
def test_viewer_replace(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer()
viewer.load(table)
viewer.replace({"a": [4, 5, 6]})
assert viewer.table.size() == 3
assert viewer.table.schema() == {"a": int}
assert viewer.table.view().to_dict() == {"a": [4, 5, 6]}
def test_viewer_delete_without_table(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer(plugin="X Bar", filter=[["a", "==", 2]])
viewer.load(table)
assert viewer.filter == [["a", "==", 2]]
viewer.delete(delete_table=False)
assert viewer.table_name is not None
assert viewer.table is not None
assert viewer.filter == []
def test_save_restore(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer(plugin="x_bar", filters=[["a", "==", 2]])
viewer.load(table)
# Save config
config = viewer.save()
assert viewer.filters == [["a", "==", 2]]
assert config["filters"] == [["a", "==", 2]]
assert viewer.plugin == "x_bar"
assert config["plugin"] == "x_bar"
# reset configuration
viewer.reset()
assert viewer.plugin == "hypergrid"
assert viewer.filters == []
# restore configuration
viewer.restore(**config)
assert viewer.filters == [["a", "==", 2]]
assert viewer.plugin == "x_bar"
def test_viewer_update_dict_partial(self):
table = Table({"a": [1, 2, 3], "b": [5, 6, 7]}, index="a")
viewer = PerspectiveViewer()
viewer.load(table)
viewer.update({"a": [1, 2, 3], "b": [8, 9, 10]})
assert table.size() == 3
assert viewer.table.size() == 3
assert viewer.table.view().to_dict() == {
"a": [1, 2, 3],
"b": [8, 9, 10]
}
def test_save_restore_plugin_config(self):
viewer = PerspectiveViewer(
plugin="Datagrid", plugin_config={"columns": {
"a": {
"fixed": 4
}
}})
config = viewer.save()
assert config["plugin_config"] == {"columns": {"a": {"fixed": 4}}}
viewer.reset()
assert viewer.plugin_config == {}
viewer.restore(**config)
assert viewer.plugin_config == config["plugin_config"]
def test_plugin_viewer_init_all(self):
for plugin in Plugin:
viewer = PerspectiveViewer(plugin=plugin)
assert viewer.plugin == plugin.value
def test_viewer_load_named_data(self):
viewer = PerspectiveViewer()
viewer.load({"a": [1, 2, 3]}, name="data_1")
assert viewer.columns == ["a"]
assert viewer.table_name == "data_1"
def test_viewer_load_data(self):
viewer = PerspectiveViewer()
viewer.load({"a": [1, 2, 3]})
assert viewer.columns == ["a"]
def test_plugin_viewer_set_all(self):
viewer = PerspectiveViewer()
for plugin in Plugin:
viewer.plugin = plugin
assert viewer.plugin == plugin.value
def test_viewer_load_table(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer()
viewer.load(table)
assert viewer.columns == ["a"]
def test_viewer_get_table(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer()
viewer.load(table)
assert viewer.table == table
def test_viewer_load_np(self):
table = Table({"a": np.arange(1, 100)})
viewer = PerspectiveViewer()
viewer.load(table)
assert viewer.columns == ["a"]
def test_viewer_load_data_with_options(self):
viewer = PerspectiveViewer()
# options should be forwarded to the Table constructor
viewer.load({"a": [1, 2, 3]}, limit=1)
assert viewer.columns == ["a"]
assert viewer.table.size() == 1
def test_viewer_make_view_none(self):
table = Table({"a": [1, 2, 3]})
viewer = PerspectiveViewer(filters=[["a", "==", 2]])
viewer.load(table)
assert viewer.view_name is None
assert viewer.view is None
def test_viewer_load_np_data(self):
viewer = PerspectiveViewer()
viewer.load({"a": np.arange(1, 100)})
assert viewer.columns == ["a"]
assert viewer.table.size() == 99
def test_viewer_load_df_data(self):
viewer = PerspectiveViewer()
viewer.load(pd.DataFrame({"a": np.arange(1, 100)}))
for col in viewer.columns:
assert col in ["index", "a"]
