__author__ = 'schelle'

import unittest
import logging
import sys
sys.path = ['../wflow'] + ['../'] + sys.path
import wflow_bmi_combined_mp as bmi
import time
import os

"""
Simple test for wflow bmi framework
"""


class MyTest(unittest.TestCase):

    def testbmifuncs(self):

        bmiobj = bmi.wflowbmi_csdms()
        bmiobj.initialize('bmirunner.ini')

        print bmiobj.get_component_name().split(',')
        print bmiobj.get_input_var_names()
        print bmiobj.get_output_var_names()
        print bmiobj.get_start_time()
        print bmiobj.get_end_time()
        print bmiobj.get_current_time()
        print bmiobj.get_time_step()
        print bmiobj.get_attribute_names()
        steps =  (bmiobj.get_end_time() -  bmiobj.get_start_time())/bmiobj.get_time_step() + 1
        print steps
        for a in range(0,steps):
            bmiobj.update()

        atn = bmiobj.get_attribute_names()
        print atn[0]
        print bmiobj.get_attribute_value(atn[0])
        bmiobj.finalize()


if __name__ == '__main__':
    unittest.main()