__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()