Commit last-minute

hydroshoot/test/non_regression_data.py

@@ -0,0 +1,46 @@
+""" 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='.')