import matplotlib.pyplot as plt
import numpy as np
from regressioninc.math import geometric_median
arr = np.array([3 - 2j, -4 - 9j, 2 + 1j, -3 + 6j])
med = geometric_median(arr)
print(np.round(med, 6))
# Expected:
## (1.444444+0.074074j)
plt.figure() # doctest: +SKIP
plt.scatter(arr.real, arr.imag, c="b", marker="x") # doctest: +SKIP
plt.scatter(med.real, med.imag, c="r", marker="o") # doctest: +SKIP
plt.tight_layout() # doctest: +SKIP
plt.show() # doctest: +SKIP