def test_weighted_median(self): out1 = sm.weighted_median(self.d, self.w1) out2 = sm.weighted_median(self.d, self.w2) self.assertEquals(out1, 4) self.assertEquals(out2, 3.5)