import pytest

import numpy as np
import pandas as pd

from bokeh.core.has_props import HasProps

import bokeh.core.property.bases as pb

def test_property_assert_bools():
    hp = HasProps()
    p = pb.Property()

    p.asserts(True, "true")
    assert p.prepare_value(hp, "foo", 10) == 10

    p.asserts(False, "false")
    with pytest.raises(ValueError) as e:
        p.prepare_value(hp, "foo", 10)
        assert str(e) == "false"

def test_property_assert_functions():
    hp = HasProps()
    p = pb.Property()

    p.asserts(lambda obj, value: True, "true")
    p.asserts(lambda obj, value: obj is hp, "true")
    p.asserts(lambda obj, value: value==10, "true")
    assert p.prepare_value(hp, "foo", 10) == 10

    p.asserts(lambda obj, value: False, "false")
    with pytest.raises(ValueError) as e:
        p.prepare_value(hp, "foo", 10)
        assert str(e) == "false"

def test_property_assert_msg_funcs():
    hp = HasProps()
    p = pb.Property()

    def raise_(ex):
        raise ex

    p.asserts(False, lambda obj, name, value: raise_(ValueError("bad %s %s %s" % (hp==obj, name, value))))

    with pytest.raises(ValueError) as e:
        p.prepare_value(hp, "foo", 10)
        assert str(e) == "bad True name, 10"

def test_property_matches_basic_types(capsys):
    p = pb.Property()
    for x in [1, 1.2, "a", np.arange(4), None, False, True, {}, []]:
        assert p.matches(x, x) is True
        assert p.matches(x, "junk") is False
    out, err = capsys.readouterr()
    assert err == ""

def test_property_matches_compatible_arrays(capsys):
    p = pb.Property()
    a = np.arange(5)
    b = np.arange(5)
    assert p.matches(a, b) is True
    assert p.matches(a, b+1) is False
    for x in [1, 1.2, "a", np.arange(4), None, False]:
        assert p.matches(a, x) is False
        assert p.matches(x, b) is False
    out, err = capsys.readouterr()
    assert err == ""

def test_property_matches_incompatible_arrays(capsys):
    p = pb.Property()
    a = np.arange(5)
    b = np.arange(5).astype(str)
    assert p.matches(a, b) is False
    out, err = capsys.readouterr()
    # no way to suppress FutureWarning in this case
    # assert err == ""

def test_property_matches_dicts_with_array_values(capsys):
    p = pb.Property()
    d1 = dict(foo=np.arange(10))
    d2 = dict(foo=np.arange(10))

    assert p.matches(d1, d1) is True
    assert p.matches(d1, d2) is True

    # XXX not sure if this is preferable to have match, or not
    assert p.matches(d1, dict(foo=list(range(10)))) is True

    assert p.matches(d1, dict(foo=np.arange(11))) is False
    assert p.matches(d1, dict(bar=np.arange(10))) is False
    assert p.matches(d1, dict(bar=10)) is False
    out, err = capsys.readouterr()
    assert err == ""

def test_property_matches_non_dict_containers_with_array_false(capsys):
    p = pb.Property()
    d1 = [np.arange(10)]
    d2 = [np.arange(10)]
    assert p.matches(d1, d1) is True  # because object identity
    assert p.matches(d1, d2) is False

    t1 = (np.arange(10),)
    t2 = (np.arange(10),)
    assert p.matches(t1, t1) is True  # because object identity
    assert p.matches(t1, t2) is False

    out, err = capsys.readouterr()
    assert err == ""

def test_property_matches_dicts_with_series_values(capsys):
    p = pb.Property()
    d1 = pd.DataFrame(dict(foo=np.arange(10)))
    d2 = pd.DataFrame(dict(foo=np.arange(10)))

    assert p.matches(d1.foo, d1.foo) is True
    assert p.matches(d1.foo, d2.foo) is True

    # XXX not sure if this is preferable to have match, or not
    assert p.matches(d1.foo, (range(10))) is True

    assert p.matches(d1.foo, np.arange(11)) is False
    assert p.matches(d1.foo, np.arange(10)+1) is False
    assert p.matches(d1.foo, 10) is False
    out, err = capsys.readouterr()
    assert err == ""

def test_property_matches_dicts_with_index_values(capsys):
    p = pb.Property()
    d1 = pd.DataFrame(dict(foo=np.arange(10)))
    d2 = pd.DataFrame(dict(foo=np.arange(10)))

    assert p.matches(d1.index, d1.index) is True
    assert p.matches(d1.index, d2.index) is True

    # XXX not sure if this is preferable to have match, or not
    assert p.matches(d1.index, list(range(10))) is True

    assert p.matches(d1.index, np.arange(11)) is False
    assert p.matches(d1.index, np.arange(10)+1) is False
    assert p.matches(d1.index, 10) is False
    out, err = capsys.readouterr()
    assert err == ""
