def ref_plot(datasets):
"""
Quickly plot a lot of datasets
Parameters
----------
datasets : iterable
strings or files identifying the datasets to plot
Returns
-------
fig : matplotlib.figure.Figure
The figure. Use fig.show() to display
"""
fig = plt.figure()
ax = fig.add_subplot(111)
for dataset in datasets:
d = ReflectDataset()
d.load(dataset)
ax.plot(d.x, d.y)
ax.autoscale(tight=True)
ax.set_yscale('log')
ax.set_xlabel(u"Q /\u212B **-1")
ax.set_ylabel('reflectivity')
return fig
def test_dot(self):
# test with file formats from http://www.reflectometry.net/refdata.htm
dataset1 = ReflectDataset()
with open(os.path.join(self.pth, "dot.dat")) as f:
dataset1.load(f)
assert_equal(len(dataset1), 31)
assert dataset1.y_err is not None
assert dataset1.x_err is None
with open(os.path.join(self.pth, "dot.aft")) as f:
dataset1.load(f)
assert_equal(len(dataset1), 36)
assert dataset1.y_err is not None
with open(os.path.join(self.pth, "dot.refl")) as f:
dataset1.load(f)
assert_equal(len(dataset1), 59)
assert dataset1.y_err is not None
with open(os.path.join(self.pth, "dot.out")) as f:
dataset1.load(f)
assert_equal(len(dataset1), 1122)
assert dataset1.y_err is not None
assert dataset1.x_err is not None
with open(os.path.join(self.pth, "dot.mft")) as f:
dataset1.load(f)
assert_equal(len(dataset1), 213)
assert dataset1.y_err is not None
assert dataset1.x_err is not None
def ref_plot(datasets):
"""
Quickly plot a lot of datasets
Parameters
----------
datasets : iterable
strings or files identifying the datasets to plot
Returns
-------
fig : matplotlib.figure.Figure
The figure. Use fig.show() to display
"""
fig = plt.figure()
ax = fig.add_subplot(111)
for dataset in datasets:
d = ReflectDataset()
d.load(dataset)
ax.plot(d.x, d.y)
ax.autoscale(tight=True)
ax.set_yscale('log')
ax.set_xlabel(u"Q /\u212B **-1")
ax.set_ylabel('reflectivity')
return fig
def test_load(self):
# load dataset from XML, via file handle
dataset = ReflectDataset()
with open(os.path.join(path, 'c_PLP0000708.xml')) as f:
dataset.load(f)
assert_equal(dataset.npoints, 90)
assert_equal(90, np.size(dataset.x))
# load dataset from XML, via string
dataset = ReflectDataset()
dataset.load(os.path.join(path, 'c_PLP0000708.xml'))
assert_equal(dataset.npoints, 90)
assert_equal(90, np.size(dataset.x))
# load dataset from .dat, via file handle
dataset1 = ReflectDataset()
with open(os.path.join(path, 'c_PLP0000708.dat')) as f:
dataset1.load(f)
assert_equal(dataset1.npoints, 90)
assert_equal(90, np.size(dataset1.x))
# load dataset from .dat, via string
dataset2 = ReflectDataset()
dataset2.load(os.path.join(path, 'c_PLP0000708.dat'))
assert_equal(dataset2.npoints, 90)
assert_equal(90, np.size(dataset2.x))
def test_load(self):
# load dataset from XML, via file handle
dataset = ReflectDataset()
with open(os.path.join(self.pth, 'c_PLP0000708.xml')) as f:
dataset.load(f)
assert_equal(len(dataset), 90)
assert_equal(90, np.size(dataset.x))
# load dataset from XML, via string
dataset = ReflectDataset()
dataset.load(os.path.join(self.pth, 'c_PLP0000708.xml'))
assert_equal(len(dataset), 90)
assert_equal(90, np.size(dataset.x))
# load dataset from .dat, via file handle
dataset1 = ReflectDataset()
with open(os.path.join(self.pth, 'c_PLP0000708.dat')) as f:
dataset1.load(f)
assert_equal(len(dataset1), 90)
assert_equal(90, np.size(dataset1.x))
# load dataset from .dat, via string
dataset2 = ReflectDataset()
dataset2.load(os.path.join(self.pth, 'c_PLP0000708.dat'))
assert_equal(len(dataset2), 90)
assert_equal(90, np.size(dataset2.x))
def test_load(self):
# test reflectivity calculation with values generated from Motofit
dataset = ReflectDataset()
with open(os.path.join(path, 'c_PLP0000708.xml')) as f:
dataset.load(f)
assert_equal(dataset.npoints, 90)
assert_equal(90, np.size(dataset.x))
dataset1 = ReflectDataset()
with open(os.path.join(path, 'c_PLP0000708.dat')) as f:
dataset1.load(f)
assert_equal(dataset1.npoints, 90)
assert_equal(90, np.size(dataset1.x))
