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