def test_load_error_with_scale(self):
raw_error = numpy.array(self.error)
data = Data()
data.path = self.error_file['filename']
data.scale = (2, 10)
data.error_columns = ((1, 2), 0)
assert_allclose(data.error[0], 2 * numpy.array([raw_error[1], raw_error[2]]))
assert_allclose(data.error[1], 10 * raw_error[0])
del data._error
data.error_columns = (0, (2, 1))
assert_allclose(data.error[0], 2 * raw_error[0])
assert_allclose(data.error[1], 10 * numpy.array([raw_error[2], raw_error[1]]))
del data._error
data.error_columns = ((1, 3), (0, 2))
assert_allclose(data.error[0], 2 * numpy.array([raw_error[1], raw_error[3]]))
assert_allclose(data.error[1], 10 * numpy.array([raw_error[0], raw_error[2]]))
def test_load_error(self):
raw_error = numpy.array(self.error)
data = Data()
data.path = self.error_file['filename']
data.error_columns = (1, None)
assert_allclose(data.error[0], raw_error[1])
eq_(data.error[1], None)
del data._error
data.error_columns = (None, 1)
eq_(data.error[0], None)
assert_allclose(data.error[1], raw_error[1])
del data._error
data.error_columns = (2, 1)
assert_allclose(data.error[0], raw_error[2])
assert_allclose(data.error[1], raw_error[1])
del data._error
data.error_columns = ((1, 2), None)
assert_allclose(data.error[0], numpy.array([raw_error[1], raw_error[2]]))
eq_(data.error[1], None)
del data._error
data.error_columns = (None, (1, 2))
eq_(data.error[0], None)
assert_allclose(data.error[1], numpy.array([raw_error[1], raw_error[2]]))
del data._error
data.error_columns = ((1, 2), 0)
assert_allclose(data.error[0], numpy.array([raw_error[1], raw_error[2]]))
assert_allclose(data.error[1], raw_error[0])
del data._error
data.error_columns = (0, (2, 1))
assert_allclose(data.error[0], raw_error[0])
assert_allclose(data.error[1], numpy.array([raw_error[2], raw_error[1]]))
del data._error
data.error_columns = ((1, 3), (0, 2))
assert_allclose(data.error[0], numpy.array([raw_error[1], raw_error[3]]))
assert_allclose(data.error[1], numpy.array([raw_error[0], raw_error[2]]))
import os
from example_helper import save_example_fit
from scipy_data_fitting import Data, Model, Fit
#
# Example of a basic linear fit with error bars.
#
name = 'linear'
# Load data from a csv file.
data = Data(name)
data.path = os.path.join('examples', 'data', 'linear.csv')
data.genfromtxt_args['skip_header'] = 1
data.error_columns = (2, 3)
# Create a linear model.
model = Model(name)
model.add_symbols('t', 'v', 'x_0')
t, v, x_0 = model.get_symbols('t', 'v', 'x_0')
model.expressions['line'] = v * t + x_0
# Create the fit using the data and model.
fit = Fit(name, data=data, model=model)
fit.expression = 'line'
fit.independent = {'symbol': 't', 'name': 'Time', 'units': 's'}
fit.dependent = {'name': 'Distance', 'units': 'm'}
fit.parameters = [
{'symbol': 'v', 'guess': 1, 'units': 'm/s'},
{'symbol': 'x_0', 'value': 1, 'units': 'm'},
