import matplotlib.pyplot as plt
import numpy as np
from regressioninc.math import geometric_median
arr = np.array([1 - 1j, -1 - 1j, -1 + 1j, 1 + 1j])
med = geometric_median(arr)
print(np.round(med, 6))
# Expected:
## 0j
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