utest_manipulations.py @ 1b28ff5

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload

Last change on this file since 1b28ff5 was e89bb46, checked in by Mathieu Doucet <doucetm@…>, 12 years ago
Consolidate duplicated tests and protect against missing deps.
Property mode set to `100644`
File size: 12.5 KB

Rev	Line
[0d86fecb]	1	"""
	2	Unit tests for data manipulations
	3	"""
	4	#TODO: what happens if you add a Data1D to a Data2D?
	5
	6	import unittest
	7	import numpy, math
[117a1d6]	8	from sans.dataloader.loader import Loader
	9	from sans.guiframe.dataFitting import Data1D, Data2D
	10	from sans.guiframe.dataFitting import Data1D as Theory1D
[0d86fecb]	11
	12	import os.path
	13
	14	class data_info_tests(unittest.TestCase):
	15
	16	def setUp(self):
	17	self.data = Loader().load("cansas1d.xml")
	18
	19	def test_clone1D(self):
	20	"""
	21	Test basic cloning
	22	"""
	23	clone = self.data.clone_without_data()
	24
	25	for i in range(len(self.data.detector)):
	26	self.assertEqual(self.data.detector[i].distance, clone.detector[i].distance)
	27
	28	class theory1d_tests(unittest.TestCase):
	29
	30	def setUp(self):
	31	self.data = Loader().load("cansas1d.xml")
	32
	33	def test_clone_theory1D(self):
	34	"""
	35	Test basic cloning
	36	"""
	37	theory = Theory1D(x=[], y=[], dy=None)
	38	theory.clone_without_data(clone=self.data)
	39	theory.copy_from_datainfo(data1d=self.data)
	40	for i in range(len(self.data.detector)):
	41	self.assertEqual(self.data.detector[i].distance, theory.detector[i].distance)
	42
	43	for i in range(len(self.data.x)):
	44	self.assertEqual(self.data.x[i], theory.x[i])
	45	self.assertEqual(self.data.y[i], theory.y[i])
	46	self.assertEqual(self.data.dy[i], theory.dy[i])
	47
	48	class manip_tests(unittest.TestCase):
	49
	50	def setUp(self):
	51	# Create two data sets to play with
	52	x_0 = numpy.ones(5)
	53	for i in range(5):
	54	x_0[i] = x_0[i]*(i+1.0)
	55
	56	y_0 = 2.0*numpy.ones(5)
	57	dy_0 = 0.5*numpy.ones(5)
	58	self.data = Data1D(x_0, y_0, dy=dy_0)
	59
	60	x = self.data.x
	61	y = numpy.ones(5)
	62	dy = numpy.ones(5)
	63	self.data2 = Data1D(x, y, dy=dy)
	64
	65
	66	def test_load(self):
	67	"""
	68	Test whether the test file was loaded properly
	69	"""
	70	# There should be 5 entries in the file
	71	self.assertEqual(len(self.data.x), 5)
	72
	73	for i in range(5):
	74	# The x values should be from 1 to 5
	75	self.assertEqual(self.data.x[i], float(i+1))
	76
	77	# All y-error values should be 0.5
	78	self.assertEqual(self.data.dy[i], 0.5)
	79
	80	# All y values should be 2.0
	81	self.assertEqual(self.data.y[i], 2.0)
	82
	83	def test_add(self):
	84	result = self.data2+self.data
	85	for i in range(5):
	86	self.assertEqual(result.y[i], 3.0)
	87	self.assertEqual(result.dy[i], math.sqrt(0.5**2+1.0))
	88
	89	def test_sub(self):
	90	result = self.data2-self.data
	91	for i in range(5):
	92	self.assertEqual(result.y[i], -1.0)
	93	self.assertEqual(result.dy[i], math.sqrt(0.5**2+1.0))
	94
	95	def test_mul(self):
	96	result = self.data2*self.data
	97	for i in range(5):
	98	self.assertEqual(result.y[i], 2.0)
	99	self.assertEqual(result.dy[i], math.sqrt((0.51.0)2+(1.02.0)**2))
	100
	101	def test_div(self):
	102	result = self.data2/self.data
	103	for i in range(5):
	104	self.assertEqual(result.y[i], 0.5)
	105	self.assertEqual(result.dy[i], math.sqrt((1.0/2.0)*2+(0.51.0/4.0)**2))
	106
	107	def test_radd(self):
	108	result = self.data+3.0
	109	for i in range(5):
	110	self.assertEqual(result.y[i], 5.0)
	111	self.assertEqual(result.dy[i], 0.5)
	112
	113	result = 3.0+self.data
	114	for i in range(5):
	115	self.assertEqual(result.y[i], 5.0)
	116	self.assertEqual(result.dy[i], 0.5)
	117
	118	def test_rsub(self):
	119	result = self.data-3.0
	120	for i in range(5):
	121	self.assertEqual(result.y[i], -1.0)
	122	self.assertEqual(result.dy[i], 0.5)
	123
	124	result = 3.0-self.data
	125	for i in range(5):
	126	self.assertEqual(result.y[i], 1.0)
	127	self.assertEqual(result.dy[i], 0.5)
	128
	129	def test_rmul(self):
	130	result = self.data*3.0
	131	for i in range(5):
	132	self.assertEqual(result.y[i], 6.0)
	133	self.assertEqual(result.dy[i], 1.5)
	134
	135	result = 3.0*self.data
	136	for i in range(5):
	137	self.assertEqual(result.y[i], 6.0)
	138	self.assertEqual(result.dy[i], 1.5)
	139
	140	def test_rdiv(self):
	141	result = self.data/4.0
	142	for i in range(5):
	143	self.assertEqual(result.y[i], 0.5)
	144	self.assertEqual(result.dy[i], 0.125)
	145
	146	result = 6.0/self.data
	147	for i in range(5):
	148	self.assertEqual(result.y[i], 3.0)
	149	self.assertEqual(result.dy[i], 6.0*0.5/4.0)
	150
	151	class manip_2D(unittest.TestCase):
	152
	153	def setUp(self):
	154	# Create two data sets to play with
[4a47244]	155	x_0 = 2.0*numpy.ones(25)
	156	dx_0 = 0.5*numpy.ones(25)
	157	qx_0 = numpy.arange(25)
	158	qy_0 = numpy.arange(25)
	159	mask_0 = numpy.zeros(25)
	160	dqx_0 = numpy.arange(25)/100
	161	dqy_0 = numpy.arange(25)/100
	162	q_0 = numpy.sqrt(qx_0 * qx_0 + qy_0 * qy_0)
	163	self.data = Data2D(image=x_0, err_image=dx_0, qx_data=qx_0,
	164	qy_data=qy_0, q_data=q_0, mask=mask_0,
	165	dqx_data=dqx_0, dqy_data=dqy_0)
[0d86fecb]	166
[4a47244]	167	y = numpy.ones(25)
	168	dy = numpy.ones(25)
	169	qx = numpy.arange(25)
	170	qy = numpy.arange(25)
	171	mask = numpy.zeros(25)
	172	q = numpy.sqrt(qx * qx + qy * qy)
	173	self.data2 = Data2D(image=y, err_image=dy, qx_data=qx, qy_data=qy,
	174	q_data=q, mask=mask)
[0d86fecb]	175
	176
	177	def test_load(self):
	178	"""
	179	Test whether the test file was loaded properly
	180	"""
	181	# There should be 5 entries in the file
[4a47244]	182	self.assertEqual(numpy.size(self.data.data), 25)
[0d86fecb]	183
[4a47244]	184	for i in range(25):
	185	# All y-error values should be 0.5
	186	self.assertEqual(self.data.err_data[i], 0.5)
	187
	188	# All y values should be 2.0
	189	self.assertEqual(self.data.data[i], 2.0)
[0d86fecb]	190
	191	def test_add(self):
	192	result = self.data2+self.data
[4a47244]	193	for i in range(25):
	194	self.assertEqual(result.data[i], 3.0)
	195	self.assertEqual(result.err_data[i], math.sqrt(0.5**2+1.0))
[0d86fecb]	196
	197	def test_sub(self):
	198	result = self.data2-self.data
[4a47244]	199	for i in range(25):
	200	self.assertEqual(result.data[i], -1.0)
	201	self.assertEqual(result.err_data[i], math.sqrt(0.5**2+1.0))
[0d86fecb]	202
	203	def test_mul(self):
	204	result = self.data2*self.data
[4a47244]	205	for i in range(25):
	206	self.assertEqual(result.data[i], 2.0)
	207	self.assertEqual(result.err_data[i], math.sqrt((0.51.0)2+(1.02.0)**2))
[0d86fecb]	208
	209	def test_div(self):
	210	result = self.data2/self.data
[4a47244]	211	for i in range(25):
	212	self.assertEqual(result.data[i], 0.5)
	213	self.assertEqual(result.err_data[i], math.sqrt((1.0/2.0)*2+(0.51.0/4.0)**2))
[0d86fecb]	214
	215	def test_radd(self):
	216	result = self.data+3.0
[4a47244]	217	for i in range(25):
	218	self.assertEqual(result.data[i], 5.0)
	219	self.assertEqual(result.err_data[i], 0.5)
	220
[0d86fecb]	221	result = 3.0+self.data
[4a47244]	222	for i in range(25):
	223	self.assertEqual(result.data[i], 5.0)
	224	self.assertEqual(result.err_data[i], 0.5)
[0d86fecb]	225
	226	def test_rsub(self):
	227	result = self.data-3.0
[4a47244]	228	for i in range(25):
	229	self.assertEqual(result.data[i], -1.0)
	230	self.assertEqual(result.err_data[i], 0.5)
	231
[0d86fecb]	232	result = 3.0-self.data
[4a47244]	233	for i in range(25):
	234	self.assertEqual(result.data[i], 1.0)
	235	self.assertEqual(result.err_data[i], 0.5)
[0d86fecb]	236
	237	def test_rmul(self):
	238	result = self.data*3.0
[4a47244]	239	for i in range(25):
	240	self.assertEqual(result.data[i], 6.0)
	241	self.assertEqual(result.err_data[i], 1.5)
	242
[0d86fecb]	243	result = 3.0*self.data
[4a47244]	244	for i in range(25):
	245	self.assertEqual(result.data[i], 6.0)
	246	self.assertEqual(result.err_data[i], 1.5)
[0d86fecb]	247
	248	def test_rdiv(self):
	249	result = self.data/4.0
[4a47244]	250	for i in range(25):
	251	self.assertEqual(result.data[i], 0.5)
	252	self.assertEqual(result.err_data[i], 0.125)
	253
[0d86fecb]	254	result = 6.0/self.data
[4a47244]	255	for i in range(25):
	256	self.assertEqual(result.data[i], 3.0)
	257	self.assertEqual(result.err_data[i], 6.0*0.5/4.0)
	258
[e89bb46]	259	class extra_manip_2D(unittest.TestCase):
	260
	261	def setUp(self):
	262	# Create two data sets to play with
	263	x_0 = 2.0*numpy.ones(25)
	264	dx_0 = 0.5*numpy.ones(25)
	265	qx_0 = numpy.arange(25)
	266	qy_0 = numpy.arange(25)
	267	mask_0 = numpy.zeros(25)
	268	dqx_0 = numpy.arange(25)/100
	269	dqy_0 = numpy.arange(25)/100
	270	q_0 = numpy.sqrt(qx_0 * qx_0 + qy_0 * qy_0)
	271	self.data = Data2D(image=x_0, err_image=dx_0, qx_data=qx_0,
	272	qy_data=qy_0, q_data=q_0, mask=mask_0,
	273	dqx_data=dqx_0, dqy_data=dqy_0)
	274
	275	y = numpy.ones(25)
	276	dy = numpy.ones(25)
	277	qx = numpy.arange(25)
	278	qy = numpy.arange(25)
	279	mask = numpy.zeros(25)
	280	q = numpy.sqrt(qx * qx + qy * qy)
	281	self.data2 = Data2D(image=y, err_image=dy, qx_data=qx, qy_data=qy,
	282	q_data=q, mask=mask)
	283
	284
	285	def test_load(self):
	286	"""
	287	Test whether the test file was loaded properly
	288	"""
	289	# There should be 5 entries in the file
	290	self.assertEqual(numpy.size(self.data.data), 25)
	291
	292	for i in range(25):
	293	# All y-error values should be 0.5
	294	self.assertEqual(self.data.err_data[i], 0.5)
	295
	296	# All y values should be 2.0
	297	self.assertEqual(self.data.data[i], 2.0)
	298
	299	def test_add(self):
	300	result = self.data2+self.data
	301	for i in range(25):
	302	self.assertEqual(result.data[i], 3.0)
	303	self.assertEqual(result.err_data[i], math.sqrt(0.5**2+1.0))
	304
	305	def test_sub(self):
	306	result = self.data2-self.data
	307	for i in range(25):
	308	self.assertEqual(result.data[i], -1.0)
	309	self.assertEqual(result.err_data[i], math.sqrt(0.5**2+1.0))
	310
	311	def test_mul(self):
	312	result = self.data2*self.data
	313	for i in range(25):
	314	self.assertEqual(result.data[i], 2.0)
	315	self.assertEqual(result.err_data[i], math.sqrt((0.51.0)2+(1.02.0)**2))
	316
	317	def test_div(self):
	318	result = self.data2/self.data
	319	for i in range(25):
	320	self.assertEqual(result.data[i], 0.5)
	321	self.assertEqual(result.err_data[i], math.sqrt((1.0/2.0)*2+(0.51.0/4.0)**2))
	322
	323	def test_radd(self):
	324	result = self.data+3.0
	325	for i in range(25):
	326	self.assertEqual(result.data[i], 5.0)
	327	self.assertEqual(result.err_data[i], 0.5)
	328
	329	result = 3.0+self.data
	330	for i in range(25):
	331	self.assertEqual(result.data[i], 5.0)
	332	self.assertEqual(result.err_data[i], 0.5)
	333
	334	def test_rsub(self):
	335	result = self.data-3.0
	336	for i in range(25):
	337	self.assertEqual(result.data[i], -1.0)
	338	self.assertEqual(result.err_data[i], 0.5)
	339
	340	result = 3.0-self.data
	341	for i in range(25):
	342	self.assertEqual(result.data[i], 1.0)
	343	self.assertEqual(result.err_data[i], 0.5)
	344
	345	def test_rmul(self):
	346	result = self.data*3.0
	347	for i in range(25):
	348	self.assertEqual(result.data[i], 6.0)
	349	self.assertEqual(result.err_data[i], 1.5)
	350
	351	result = 3.0*self.data
	352	for i in range(25):
	353	self.assertEqual(result.data[i], 6.0)
	354	self.assertEqual(result.err_data[i], 1.5)
	355
	356	def test_rdiv(self):
	357	result = self.data/4.0
	358	for i in range(25):
	359	self.assertEqual(result.data[i], 0.5)
	360	self.assertEqual(result.err_data[i], 0.125)
	361
	362	result = 6.0/self.data
	363	for i in range(25):
	364	self.assertEqual(result.data[i], 3.0)
	365	self.assertEqual(result.err_data[i], 6.0*0.5/4.0)
[0d86fecb]	366
	367	if __name__ == '__main__':
	368	unittest.main()
	369

Note: See TracBrowser for help on using the repository browser.

SasView

source: sasview/sansguiframe/test/utest_manipulations.py @ 1b28ff5

Download in other formats: