from extend_template_method import *


em = ExtendMe()

ret_double = em.do_stuff_double(1, 1.1)
if ret_double != 1.1:
    raise RuntimeError("double failed " + ret_double)
ret_string = em.do_stuff_string(1, "hello there")
if ret_string != "hello there":
    raise RuntimeError("string failed " + ret_string)

ret_double = em.do_overloaded_stuff(1.1)
if ret_double != 1.1:
    raise RuntimeError("double failed " + ret_double)
ret_string = em.do_overloaded_stuff("hello there")
if ret_string != "hello there":
    raise RuntimeError("string failed " + ret_string)


em = TemplateExtend()

ret_double = em.do_template_stuff_double(1, 1.1)
if ret_double != 1.1:
    raise RuntimeError("double failed " + ret_double)
ret_string = em.do_template_stuff_string(1, "hello there")
if ret_string != "hello there":
    raise RuntimeError("string failed " + ret_string)


ret_double = em.do_template_overloaded_stuff(1.1)
if ret_double != 1.1:
    raise RuntimeError("double failed " + ret_double)
ret_string = em.do_template_overloaded_stuff("hello there")
if ret_string != "hello there":
    raise RuntimeError("string failed " + ret_string)