def test_irwls_1d(self): z = IRWLS.irwls(self.x, self.y, self.update_func, 2, self.w) assert_array_equal(z.est.shape, (1, 1)) assert_array_almost_equal(z.est, 1)