def test_add_density(self): p = np.arange(10) t = 20.0 * 0.2 * p s = 30.0 * 0.25 * p x = [-20.0 for _ in p] y = [50.0 for _ in p] sa = gsw.sa_from_sp(s, p, x, y) ct = gsw.ct_from_t(sa, t, p) rho = gsw.rho(sa, ct, p) cast = CTDCast(p, s, t, coordinates=(-20, 50)) cast.add_density() self.assertTrue(np.allclose(rho, cast["density"])) return