""" A collection of  data objects to be used for testing"""

from os.path import dirname, join
from json import load
from pandas import read_csv, DatetimeIndex

from openalea.mtg import traversal
from hydroshoot import architecture

sources_dir = join(dirname(__file__), 'data')


def potted_syrah():
    """Returns an `openalea.mtg` representing a potted syrah grapevine."""
    digit = join(sources_dir, 'grapevine_pot.csv')
    g = architecture.vine_mtg(digit)
    for v in traversal.iter_mtg2(g, g.root):
        architecture.vine_phyto_modular(g, v)
        architecture.vine_mtg_properties(g, v)
        architecture.vine_mtg_geometry(g, v)
        architecture.vine_transform(g, v)

    return g


def meteo():
    """Returns a `pandas.DataFrame` containing meteorological data."""
    meteo_path = join(sources_dir, 'meteo.csv')
    df = read_csv(meteo_path, sep=';', decimal='.', header=0)
    df.time = DatetimeIndex(df.time)
    df = df.set_index(df.time)
    return df


def json_parameters():
    """Returns `Dict` of hydroshoot parameters."""
    params_path = join(sources_dir, 'params.json')
    with open(params_path) as f:
        pars = load(f)
    return pars


def reference_time_series_output():
    """Returns a `pandas.DataFrame` containing reference hydroshoot time-series output."""
    path = join(sources_dir, 'reference_time_series.csv')
    return read_csv(path, sep=';', decimal='.')