circuitpython/tests/float/complex1.py at master · ProGamerCode/circuitpython · GitHub

History

116 lines (98 loc) · 2.18 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

# test basic complex number functionality

# constructor

print(complex(1))

print(complex(1.2))

print(complex(1.2j))

print(complex("1"))

print(complex("1.2"))

print(complex("1.2j"))

print(complex(1, 2))

print(complex(1j, 2j))

# unary ops

print(bool(1j))

print(+(1j))

print(-(1 + 2j))

# binary ops

print(1j + False)

print(1j + True)

print(1j + 2)

print(1j + 2j)

print(1j - 2)

print(1j - 2j)

print(1j * 2)

print(1j * 2j)

print(1j / 2)

print((1j / 2j).real)

print(1j / (1 + 2j))

ans = 0j ** 0; print("%.5g %.5g" % (ans.real, ans.imag))

ans = 0j ** 1; print("%.5g %.5g" % (ans.real, ans.imag))

ans = 0j ** 0j; print("%.5g %.5g" % (ans.real, ans.imag))

ans = 1j ** 2.5; print("%.5g %.5g" % (ans.real, ans.imag))

ans = 1j ** 2.5j; print("%.5g %.5g" % (ans.real, ans.imag))

# comparison

print(1j == 1)

print(1j == 1j)

# comparison of nan is special

nan = float('nan') * 1j

print(nan == 1j)

print(nan == nan)

# builtin abs

print(abs(1j))

print("%.5g" % abs(1j + 2))

# builtin hash

print(hash(1 + 0j))

print(type(hash(1j)))

# float on lhs should delegate to complex

print(1.2 + 3j)

# negative base and fractional power should create a complex

ans = (-1) ** 2.3; print("%.5g %.5g" % (ans.real, ans.imag))

ans = (-1.2) ** -3.4; print("%.5g %.5g" % (ans.real, ans.imag))

# check printing of inf/nan

print(float('nan') * 1j)

print(float('-nan') * 1j)

print(float('inf') * (1 + 1j))

print(float('-inf') * (1 + 1j))

# can't assign to attributes

try:

(1j).imag = 0

except AttributeError:

print('AttributeError')

# can't convert rhs to complex

try:

1j + []

except TypeError:

print("TypeError")

# unsupported unary op

try:

~(1j)

except TypeError:

print("TypeError")

# unsupported binary op

try:

1j // 2

except TypeError:

print("TypeError")

# unsupported binary op

try:

1j < 2j

except TypeError:

print("TypeError")

#small int on LHS, complex on RHS, unsupported op

try:

print(1 | 1j)

except TypeError:

print('TypeError')

# zero division

try:

1j / 0

except ZeroDivisionError:

print("ZeroDivisionError")

# zero division via power

try:

0j ** -1

except ZeroDivisionError:

print("ZeroDivisionError")

try:

0j ** 1j

except ZeroDivisionError:

print("ZeroDivisionError")

circuitpython/tests/float/complex1.py at master · ProGamerCode/circuitpython · GitHub

Expand file tree

Breadcrumbs

complex1.py

Latest commit

History

Breadcrumbs

complex1.py

File metadata and controls