def test_transform(self):
pth = os.path.dirname(os.path.abspath(__file__))
fname = os.path.join(pth, 'c_PLP0011859_q.txt')
data = ReflectDataset(fname)
t = Transform('logY')
yt, et = t(data.x, data.y, y_err=data.y_err)
assert_equal(yt, np.log10(data.y))
yt, _ = t(data.x, data.y, y_err=None)
assert_equal(yt, np.log10(data.y))
EPy, EPe = EP.EPlog10(data.y, data.y_err)
assert_equal(yt, EPy)
assert_equal(et, EPe)
def __transform(self, x, y, y_err=None):
r"""
Transform the data passed in
Parameters
----------
x : array-like
y : array-like
y_err : array-like
Returns
-------
yt, et : tuple
The transformed data
"""
if y_err is None:
etemp = np.ones_like(y)
else:
etemp = y_err
if self.form in ["lin", None]:
yt = np.copy(y)
et = np.copy(etemp)
elif self.form == "logY":
yt, et = EP.EPlog10(y, etemp)
if not np.isfinite(yt).all():
warnings.warn(
"Some of the transformed data was non-finite."
" Please check your datasets for points with zero or"
" negative values.",
RuntimeWarning,
)
elif self.form == "YX4":
yt = y * np.power(x, 4)
et = etemp * np.power(x, 4)
elif self.form == "YX2":
yt = y * np.power(x, 2)
et = etemp * np.power(x, 2)
if y_err is None:
return yt, None
else:
return yt, et
def __transform(self, x, y, y_err=None):
r"""
Transform the data passed in
Parameters
----------
x : array-like
y : array-like
y_err : array-like
Returns
-------
yt, et : tuple
The transformed data
"""
if y_err is None:
etemp = np.ones_like(y)
else:
etemp = y_err
if self.form in ['lin', None]:
yt = np.copy(y)
et = np.copy(etemp)
elif self.form == 'logY':
yt, et = EP.EPlog10(y, etemp)
elif self.form == 'YX4':
yt = y * np.power(x, 4)
et = etemp * np.power(x, 4)
elif self.form == 'YX2':
yt = y * np.power(x, 2)
et = etemp * np.power(x, 2)
if y_err is None:
return yt, None
else:
return yt, et
def test_log10(self):
c, dc = EP.EPlog10(1.1, 0.5)
assert_array_almost_equal(0.04139268515822508, c)
assert_array_almost_equal(0.19740658268329625, dc)
