Update tests

Include test for complex numbers