def test_Disk_Smoother(self):
self.kw.transform = 'b'
exp = [
0.12222222000000001, 0.10833333, 0.08055556, 0.08944444,
0.09944444, 0.09351852
]
disk = sm.Disk_Smoother(self.e, self.b, self.kw)
self.assertEqual(list(disk.r.round(8)), exp)
def test_Disk_Smoother_tabular(self):
self.kw.transform = 'b'
dfa = self.dfa
disk = sm.Disk_Smoother(dfa.e, dfa.b, self.kw).r
np.testing.assert_allclose(disk.flatten(), self.disk_exp)
disk = sm.Disk_Smoother.by_col(dfa, 'e', 'b', self.kw)
col = 'e-b_disk_smoother'
np.testing.assert_allclose(disk[col].values,
self.disk_exp,
rtol=RTOL,
atol=ATOL)
def test_Disk_Smoother(self):
self.kw.transform = 'b'
disk = sm.Disk_Smoother(self.e, self.b, self.kw)
np.testing.assert_allclose(disk.r.flatten(), self.disk_exp)