Example #1
0
def test_type_detect_series():
    datetime_series = pd.Series([
        datetime.datetime(2010, 1, 1),
        datetime.datetime(2010, 8, 2),
        datetime.datetime(2011, 2, 1),
        np.datetime64("NaT"),
    ])

    typeset = StandardSet()
    detected_type = detect_series_type(datetime_series, typeset)
    assert detected_type == DateTime
def generate_typeset_plots() -> None:
    typesets_dir = Path("typesets/")
    typesets_dir.mkdir(exist_ok=True)

    # Initialize typeset
    for name, tsc in [
        ("typeset_complete", CompleteSet()),
        ("typeset_geometry", GeometrySet()),
        ("typeset_standard", StandardSet()),
    ]:
        # Write graph to dot
        tsc.output_graph(typesets_dir / "{name}.dot".format(name=name))

        # Plot the graph (svg)
        tsc.output_graph(typesets_dir / "{name}.svg".format(name=name))
        tsc.output_graph(typesets_dir / "{name}_base.svg".format(name=name),
                         base_only=True)

        # Plot the graph (png)
        tsc.output_graph(typesets_dir / "{name}.png".format(name=name),
                         dpi=150)
Example #3
0
import pandas as pd

from visions.functional import detect_series_type
from visions.typesets import StandardSet
from visions.types import Categorical, Boolean

from examples.data_analysis.categorical import Category

ts = StandardSet()
ts -= Boolean
ts -= Categorical
ts += Category

s1 = pd.Series(["A", "B", "C"] * 1000, dtype="category")
print(s1 in Category)
print(detect_series_type(s1, ts))

s2 = pd.Series([True, False] * 1000)
print(s2 in Category)
print(detect_series_type(s2, ts))
def test_type_cast_detect_series():
    string_series = pd.Series(["(12.0+10.0j)", "(-4.0+6.2j)", "(8.0+2.0j)"])

    typeset = StandardSet()
    new_series = cast_to_detected(string_series, typeset)
    assert new_series.iloc[1] == "(-4.0+6.2j)"
def test_type_cast_infer_series():
    string_series = pd.Series(["(12.0+10.0j)", "(-4.0+6.2j)", "(8.0+2.0j)"])

    typeset = StandardSet()
    new_series = cast_to_inferred(string_series, typeset)
    assert new_series.iloc[1].real == -4.0
def test_type_inference_series():
    string_series = pd.Series(["(12.0+10.0j)", "(-4.0+6.2j)", "(8.0+2.0j)"])

    typeset = StandardSet()
    detected_type = infer_type(string_series, typeset)
    assert detected_type == Complex
Example #7
0
from pathlib import Path

from visions.typesets import CompleteSet, StandardSet, GeometrySet

# Windows Note
# Tip for Python3/64-bit compatible version of pygraphviz
# https://github.com/CristiFati/Prebuilt-Binaries/raw/master/Windows/PyGraphviz/pygraphviz-1.5-cp37-cp37m-win_amd64.whl

typesets_dir = Path("typesets/")
typesets_dir.mkdir(exist_ok=True)

# Initialize typeset
for name, tsc in [
    ("typeset_complete", CompleteSet()),
    ("typeset_geometry", GeometrySet()),
    ("typeset_standard", StandardSet()),
]:
    # Write graph to dot
    tsc.output_graph(typesets_dir / "{name}.dot".format(name=name))

    # Plot the graph (svg)
    tsc.output_graph(typesets_dir / "{name}.svg".format(name=name))
    tsc.output_graph(typesets_dir / "{name}_base.svg".format(name=name),
                     base_only=True)

    # Plot the graph (png)
    tsc.output_graph(typesets_dir / "{name}.png".format(name=name), dpi=150)