wehub-resource-sync
454 lines
15 KiB
Python
454 lines
15 KiB
Python
# ┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
|
|
# ┃ ██████ ██████ ██████ █ █ █ █ █ █▄ ▀███ █ ┃
|
|
# ┃ ▄▄▄▄▄█ █▄▄▄▄▄ ▄▄▄▄▄█ ▀▀▀▀▀█▀▀▀▀▀ █ ▀▀▀▀▀█ ████████▌▐███ ███▄ ▀█ █ ▀▀▀▀▀ ┃
|
|
# ┃ █▀▀▀▀▀ █▀▀▀▀▀ █▀██▀▀ ▄▄▄▄▄ █ ▄▄▄▄▄█ ▄▄▄▄▄█ ████████▌▐███ █████▄ █ ▄▄▄▄▄ ┃
|
|
# ┃ █ ██████ █ ▀█▄ █ ██████ █ ███▌▐███ ███████▄ █ ┃
|
|
# ┣━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┫
|
|
# ┃ Copyright (c) 2017, the Perspective Authors. ┃
|
|
# ┃ ╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌╌ ┃
|
|
# ┃ This file is part of the Perspective library, distributed under the terms ┃
|
|
# ┃ of the [Apache License 2.0](https://www.apache.org/licenses/LICENSE-2.0). ┃
|
|
# ┗━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┛
|
|
|
|
import pandas as pd
|
|
import numpy as np
|
|
import pytest
|
|
from perspective.widget import PerspectiveWidget
|
|
import perspective as psp
|
|
|
|
client = psp.Server().new_local_client()
|
|
Table = client.table
|
|
|
|
|
|
pytest.skip(allow_module_level=True)
|
|
|
|
|
|
class TestWidgetPandas:
|
|
def test_widget_load_table_df(self, superstore):
|
|
table = Table(superstore)
|
|
widget = PerspectiveWidget(table)
|
|
assert widget.table.schema() == {
|
|
"index": "integer",
|
|
"Country": "string",
|
|
"Region": "string",
|
|
"Category": "string",
|
|
"City": "string",
|
|
"Customer ID": "string",
|
|
"Discount": "float",
|
|
"Order Date": "date",
|
|
"Order ID": "string",
|
|
"Postal Code": "string",
|
|
"Product ID": "string",
|
|
"Profit": "float",
|
|
"Quantity": "integer",
|
|
"Row ID": "integer",
|
|
"Sales": "integer",
|
|
"Segment": "string",
|
|
"Ship Date": "date",
|
|
"Ship Mode": "string",
|
|
"State": "string",
|
|
"Sub-Category": "string",
|
|
}
|
|
|
|
assert sorted(widget.columns) == sorted(
|
|
[
|
|
"index",
|
|
"Category",
|
|
"City",
|
|
"Country",
|
|
"Customer ID",
|
|
"Discount",
|
|
"Order Date",
|
|
"Order ID",
|
|
"Postal Code",
|
|
"Product ID",
|
|
"Profit",
|
|
"Quantity",
|
|
"Region",
|
|
"Row ID",
|
|
"Sales",
|
|
"Segment",
|
|
"Ship Date",
|
|
"Ship Mode",
|
|
"State",
|
|
"Sub-Category",
|
|
]
|
|
)
|
|
view = widget.table.view()
|
|
assert view.num_rows() == len(superstore)
|
|
assert view.num_columns() == len(superstore.columns) + 1 # index
|
|
|
|
def test_widget_load_data_df(self, superstore):
|
|
widget = PerspectiveWidget(superstore)
|
|
assert sorted(widget.columns) == sorted(
|
|
[
|
|
"index",
|
|
"Category",
|
|
"City",
|
|
"Country",
|
|
"Customer ID",
|
|
"Discount",
|
|
"Order Date",
|
|
"Order ID",
|
|
"Postal Code",
|
|
"Product ID",
|
|
"Profit",
|
|
"Quantity",
|
|
"Region",
|
|
"Row ID",
|
|
"Sales",
|
|
"Segment",
|
|
"Ship Date",
|
|
"Ship Mode",
|
|
"State",
|
|
"Sub-Category",
|
|
]
|
|
)
|
|
view = widget.table.view()
|
|
assert view.num_rows() == len(superstore)
|
|
assert view.num_columns() == 20
|
|
|
|
def test_widget_load_series(self, superstore):
|
|
series = pd.Series(superstore["Profit"].values, name="profit")
|
|
widget = PerspectiveWidget(series)
|
|
assert widget.table.schema() == {"index": "integer", "profit": "float"}
|
|
|
|
assert sorted(widget.columns) == sorted(["index", "profit"])
|
|
view = widget.table.view()
|
|
assert view.num_rows() == len(superstore)
|
|
assert view.num_columns() == 2
|
|
|
|
def test_widget_load_pivot_table(self, superstore):
|
|
pivot_table = pd.pivot_table(
|
|
superstore,
|
|
values="Discount",
|
|
index=["Country", "Region"],
|
|
columns=["Category", "Segment"],
|
|
)
|
|
widget = PerspectiveWidget(pivot_table)
|
|
assert widget.group_by == ["Country", "Region"]
|
|
assert widget.split_by == ["Category", "Segment"]
|
|
assert widget.columns == ["value"]
|
|
# table should host flattened data
|
|
view = widget.table.view()
|
|
assert view.num_rows() == 60
|
|
assert view.num_columns() == 6
|
|
|
|
def test_widget_load_pivot_table_with_user_pivots(self, superstore):
|
|
pivot_table = pd.pivot_table(
|
|
superstore,
|
|
values="Discount",
|
|
index=["Country", "Region"],
|
|
columns="Category",
|
|
)
|
|
widget = PerspectiveWidget(pivot_table, group_by=["Category", "Segment"])
|
|
assert widget.group_by == ["Category", "Segment"]
|
|
assert widget.split_by == []
|
|
assert widget.columns == [
|
|
"index",
|
|
"Country",
|
|
"Region",
|
|
"Financials",
|
|
"Industrials",
|
|
"Technology",
|
|
]
|
|
# table should host flattened data
|
|
view = widget.table.view()
|
|
assert view.num_rows() == 5
|
|
assert view.num_columns() == 6
|
|
|
|
def test_widget_load_group_by(self, superstore):
|
|
df_pivoted = superstore.set_index(["Country", "Region"])
|
|
widget = PerspectiveWidget(df_pivoted)
|
|
assert widget.group_by == ["Country", "Region"]
|
|
assert widget.split_by == []
|
|
assert sorted(widget.columns) == sorted(
|
|
[
|
|
"index",
|
|
"Category",
|
|
"Country",
|
|
"City",
|
|
"Customer ID",
|
|
"Discount",
|
|
"Order Date",
|
|
"Order ID",
|
|
"Postal Code",
|
|
"Product ID",
|
|
"Profit",
|
|
"Quantity",
|
|
"Region",
|
|
"Row ID",
|
|
"Sales",
|
|
"Segment",
|
|
"Ship Date",
|
|
"Ship Mode",
|
|
"State",
|
|
"Sub-Category",
|
|
]
|
|
)
|
|
assert widget.table.size() == 100
|
|
view = widget.table.view()
|
|
assert view.num_rows() == len(superstore)
|
|
assert view.num_columns() == len(superstore.columns) + 1 # index
|
|
|
|
def test_widget_load_group_by_with_user_pivots(self, superstore):
|
|
df_pivoted = superstore.set_index(["Country", "Region"])
|
|
widget = PerspectiveWidget(df_pivoted, group_by=["Category", "Segment"])
|
|
assert widget.group_by == ["Category", "Segment"]
|
|
assert widget.split_by == []
|
|
assert sorted(widget.columns) == sorted(
|
|
[
|
|
"index",
|
|
"Category",
|
|
"Country",
|
|
"City",
|
|
"Customer ID",
|
|
"Discount",
|
|
"Order Date",
|
|
"Order ID",
|
|
"Postal Code",
|
|
"Product ID",
|
|
"Profit",
|
|
"Quantity",
|
|
"Region",
|
|
"Row ID",
|
|
"Sales",
|
|
"Segment",
|
|
"Ship Date",
|
|
"Ship Mode",
|
|
"State",
|
|
"Sub-Category",
|
|
]
|
|
)
|
|
assert widget.table.size() == 100
|
|
view = widget.table.view()
|
|
assert view.num_rows() == len(superstore)
|
|
assert view.num_columns() == len(superstore.columns) + 1 # index
|
|
|
|
def test_widget_load_split_by(self, superstore):
|
|
arrays = [
|
|
np.array(
|
|
[
|
|
"bar",
|
|
"bar",
|
|
"bar",
|
|
"bar",
|
|
"baz",
|
|
"baz",
|
|
"baz",
|
|
"baz",
|
|
"foo",
|
|
"foo",
|
|
"foo",
|
|
"foo",
|
|
"qux",
|
|
"qux",
|
|
"qux",
|
|
"qux",
|
|
]
|
|
),
|
|
np.array(
|
|
[
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
]
|
|
),
|
|
np.array(
|
|
[
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
]
|
|
),
|
|
]
|
|
tuples = list(zip(*arrays))
|
|
index = pd.MultiIndex.from_tuples(tuples, names=["first", "second", "third"])
|
|
df_both = pd.DataFrame(
|
|
np.random.randn(3, 16), index=["A", "B", "C"], columns=index
|
|
)
|
|
widget = PerspectiveWidget(df_both)
|
|
assert widget.columns == ["value"]
|
|
assert widget.split_by == ["first", "second", "third"]
|
|
assert widget.group_by == ["index"]
|
|
|
|
def test_widget_load_split_by_preserve_user_settings(self, superstore):
|
|
arrays = [
|
|
np.array(
|
|
[
|
|
"bar",
|
|
"bar",
|
|
"bar",
|
|
"bar",
|
|
"baz",
|
|
"baz",
|
|
"baz",
|
|
"baz",
|
|
"foo",
|
|
"foo",
|
|
"foo",
|
|
"foo",
|
|
"qux",
|
|
"qux",
|
|
"qux",
|
|
"qux",
|
|
]
|
|
),
|
|
np.array(
|
|
[
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
]
|
|
),
|
|
np.array(
|
|
[
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
]
|
|
),
|
|
]
|
|
tuples = list(zip(*arrays))
|
|
index = pd.MultiIndex.from_tuples(tuples, names=["first", "second", "third"])
|
|
df_both = pd.DataFrame(
|
|
np.random.randn(3, 16), index=["A", "B", "C"], columns=index
|
|
)
|
|
widget = PerspectiveWidget(df_both, columns=["first", "third"])
|
|
assert widget.columns == ["first", "third"]
|
|
assert widget.split_by == ["first", "second", "third"]
|
|
assert widget.group_by == ["index"]
|
|
|
|
def test_pivottable_values_index(self, superstore):
|
|
arrays = {
|
|
"A": [
|
|
"bar",
|
|
"bar",
|
|
"bar",
|
|
"bar",
|
|
"baz",
|
|
"baz",
|
|
"baz",
|
|
"baz",
|
|
"foo",
|
|
"foo",
|
|
"foo",
|
|
"foo",
|
|
"qux",
|
|
"qux",
|
|
"qux",
|
|
"qux",
|
|
],
|
|
"B": [
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
"one",
|
|
"one",
|
|
"two",
|
|
"two",
|
|
],
|
|
"C": [
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
"X",
|
|
"Y",
|
|
],
|
|
"D": np.arange(16),
|
|
}
|
|
|
|
df = pd.DataFrame(arrays)
|
|
df_pivot = df.pivot_table(
|
|
values=["D"], index=["A"], columns=["B", "C"], aggfunc={"D": "count"}
|
|
)
|
|
widget = PerspectiveWidget(df_pivot)
|
|
assert widget.columns == ["value"]
|
|
assert widget.split_by == ["B", "C"]
|
|
assert widget.group_by == ["A"]
|
|
|
|
def test_pivottable_multi_values(self, superstore):
|
|
pt = pd.pivot_table(
|
|
superstore,
|
|
values=["Discount", "Sales"],
|
|
index=["Country", "Region"],
|
|
aggfunc={"Discount": "count", "Sales": "sum"},
|
|
columns=["State", "Quantity"],
|
|
)
|
|
widget = PerspectiveWidget(pt)
|
|
assert widget.columns == ["Discount", "Sales"]
|
|
assert widget.split_by == ["State", "Quantity"]
|
|
assert widget.group_by == ["Country", "Region"]
|