163 lines
4.6 KiB
Python
163 lines
4.6 KiB
Python
from typing import List, Set, Type
|
|
|
|
import pytest
|
|
|
|
from pynecone import data
|
|
from pynecone.components.component import Component, CustomComponent, ImportDict
|
|
from pynecone.components.layout.box import Box
|
|
from pynecone.event import EVENT_TRIGGERS, EventHandler
|
|
from pynecone.state import State
|
|
from pynecone.style import Style
|
|
from pynecone.var import Var
|
|
|
|
x_num = [1, 2, 3, 4, 5]
|
|
x_str = ["Cats", "Dogs", "Birds", "Fish", "Reptiles"]
|
|
y = [1, 2, 3, 4, 10]
|
|
y1 = [5, 12, 4, 6, 1]
|
|
y2 = [
|
|
[1, 5, 7, 4, 10, 14],
|
|
[1, 2, 3, 4, 10],
|
|
[1, 2, 3, 4, 5],
|
|
[1, 7, 3, 14, 10],
|
|
[1, 2, 6, 4, 10],
|
|
]
|
|
amount = [1, 5, 3, 14, 1]
|
|
|
|
|
|
def test_line():
|
|
output = data(graph="line", x=x_num, y=y)
|
|
expected = [
|
|
{"x": 1, "y": 1},
|
|
{"x": 2, "y": 2},
|
|
{"x": 3, "y": 3},
|
|
{"x": 4, "y": 4},
|
|
{"x": 5, "y": 10},
|
|
]
|
|
assert output == expected
|
|
|
|
|
|
def test_scatter():
|
|
output = data(graph="scatter", x=x_num, y=y)
|
|
expected = [
|
|
{"x": 1, "y": 1},
|
|
{"x": 2, "y": 2},
|
|
{"x": 3, "y": 3},
|
|
{"x": 4, "y": 4},
|
|
{"x": 5, "y": 10},
|
|
]
|
|
assert output == expected
|
|
|
|
|
|
def test_area():
|
|
output = data(graph="area", x=x_num, y=y)
|
|
expected = [
|
|
{"x": 1, "y": 1},
|
|
{"x": 2, "y": 2},
|
|
{"x": 3, "y": 3},
|
|
{"x": 4, "y": 4},
|
|
{"x": 5, "y": 10},
|
|
]
|
|
assert output == expected
|
|
|
|
|
|
def test_bar():
|
|
output = data(graph="bar", x=x_str, y=y)
|
|
expected = [
|
|
{"x": "Cats", "y": 1},
|
|
{"x": "Dogs", "y": 2},
|
|
{"x": "Birds", "y": 3},
|
|
{"x": "Fish", "y": 4},
|
|
{"x": "Reptiles", "y": 10},
|
|
]
|
|
assert output == expected
|
|
|
|
|
|
def test_box_plot():
|
|
output = data(graph="box_plot", x=x_num, y=y2)
|
|
expected = [
|
|
{"x": 1, "y": [1, 5, 7, 4, 10, 14]},
|
|
{"x": 2, "y": [1, 2, 3, 4, 10]},
|
|
{"x": 3, "y": [1, 2, 3, 4, 5]},
|
|
{"x": 4, "y": [1, 7, 3, 14, 10]},
|
|
{"x": 5, "y": [1, 2, 6, 4, 10]},
|
|
]
|
|
output_specified = data(
|
|
graph="box_plot", x=x_num, min_=y1, max_=y1, median=y1, q1=y1, q3=y1
|
|
)
|
|
expected_specified = [
|
|
{"x": 1, "min": 5, "max": 5, "median": 5, "q1": 5, "q3": 5},
|
|
{"x": 2, "min": 12, "max": 12, "median": 12, "q1": 12, "q3": 12},
|
|
{"x": 3, "min": 4, "max": 4, "median": 4, "q1": 4, "q3": 4},
|
|
{"x": 4, "min": 6, "max": 6, "median": 6, "q1": 6, "q3": 6},
|
|
{"x": 5, "min": 1, "max": 1, "median": 1, "q1": 1, "q3": 1},
|
|
]
|
|
assert output == expected
|
|
assert output_specified == expected_specified
|
|
|
|
|
|
def test_histogram():
|
|
output = data(graph="histogram", x=x_num)
|
|
output2 = data(graph="histogram", x=y1)
|
|
expected = [{"x": 1}, {"x": 2}, {"x": 3}, {"x": 4}, {"x": 5}]
|
|
expected2 = [{"x": 5}, {"x": 12}, {"x": 4}, {"x": 6}, {"x": 1}]
|
|
assert output == expected
|
|
assert output2 == expected2
|
|
|
|
|
|
def test_pie():
|
|
output = data(graph="pie", x=x_str, y=amount)
|
|
expected = [
|
|
{"x": "Cats", "y": 1},
|
|
{"x": "Dogs", "y": 5},
|
|
{"x": "Birds", "y": 3},
|
|
{"x": "Fish", "y": 14},
|
|
{"x": "Reptiles", "y": 1},
|
|
]
|
|
output_labels = data(graph="pie", x=x_str, y=amount, label=x_str)
|
|
expected_labels = [
|
|
{"x": "Cats", "y": 1, "label": "Cats"},
|
|
{"x": "Dogs", "y": 5, "label": "Dogs"},
|
|
{"x": "Birds", "y": 3, "label": "Birds"},
|
|
{"x": "Fish", "y": 14, "label": "Fish"},
|
|
{"x": "Reptiles", "y": 1, "label": "Reptiles"},
|
|
]
|
|
assert output == expected
|
|
assert output_labels == expected_labels
|
|
|
|
|
|
def test_voronoi():
|
|
output = data(graph="voronoi", x=x_num, y=y)
|
|
expected = [
|
|
{"x": 1, "y": 1},
|
|
{"x": 2, "y": 2},
|
|
{"x": 3, "y": 3},
|
|
{"x": 4, "y": 4},
|
|
{"x": 5, "y": 10},
|
|
]
|
|
assert output == expected
|
|
|
|
|
|
def test_candlestick():
|
|
output = data(graph="candlestick", x=x_num, open=y1, high=y1, low=y1, close=y1)
|
|
expected = [
|
|
{"x": 1, "open": 5, "high": 5, "low": 5, "close": 5},
|
|
{"x": 2, "open": 12, "high": 12, "low": 12, "close": 12},
|
|
{"x": 3, "open": 4, "high": 4, "low": 4, "close": 4},
|
|
{"x": 4, "open": 6, "high": 6, "low": 6, "close": 6},
|
|
{"x": 5, "open": 1, "high": 1, "low": 1, "close": 1},
|
|
]
|
|
|
|
assert output == expected
|
|
|
|
|
|
def test_errorbar():
|
|
output = data(graph="error_bar", x=x_num, y=y1, error_y=y1, error_x=y1)
|
|
expected = [
|
|
{"x": 1, "y": 5, "errorY": 5, "errorX": 5},
|
|
{"x": 2, "y": 12, "errorY": 12, "errorX": 12},
|
|
{"x": 3, "y": 4, "errorY": 4, "errorX": 4},
|
|
{"x": 4, "y": 6, "errorY": 6, "errorX": 6},
|
|
{"x": 5, "y": 1, "errorY": 1, "errorX": 1},
|
|
]
|
|
assert output == expected
|