def createTestWorkspace(self):
""" Create a workspace for testing against with ideal log values
"""
from mantid.simpleapi import CreateWorkspace
from mantid.simpleapi import AddSampleLog
from time import gmtime, strftime,mktime
import numpy as np
# Create a matrix workspace
x = np.array([1.,2.,3.,4.])
y = np.array([1.,2.,3.])
e = np.sqrt(np.array([1.,2.,3.]))
wksp = CreateWorkspace(DataX=x, DataY=y,DataE=e,NSpec=1,UnitX='TOF')
# Add run_start
tmptime = strftime("%Y-%m-%d %H:%M:%S", gmtime(mktime(gmtime())))
AddSampleLog(Workspace=wksp,LogName='run_start',LogText=str(tmptime))
tsp_a=kernel.FloatTimeSeriesProperty("SensorA")
tsp_b=kernel.FloatTimeSeriesProperty("SensorB")
tsp_c=kernel.FloatTimeSeriesProperty("SensorC")
for i in arange(25):
tmptime = strftime("%Y-%m-%d %H:%M:%S", gmtime(mktime(gmtime())+i))
tsp_a.addValue(tmptime, 1.0*i*i)
tsp_b.addValue(tmptime, 2.0*i*i)
tsp_c.addValue(tmptime, 3.0*i*i)
wksp.mutableRun()['SensorA']=tsp_a
wksp.mutableRun()['SensorB']=tsp_b
wksp.mutableRun()['SensorC']=tsp_c
return wksp
def createTestWorkspace(self):
""" Create a workspace for testing against with ideal log values
"""
from mantid.simpleapi import CreateWorkspace
from mantid.simpleapi import AddSampleLog
from time import gmtime, strftime, mktime
# Create a matrix workspace
x = np.array([1., 2., 3., 4.])
y = np.array([1., 2., 3.])
e = np.sqrt(np.array([1., 2., 3.]))
wksp = CreateWorkspace(DataX=x, DataY=y, DataE=e, NSpec=1, UnitX='TOF')
# Add run_start
tmptime = strftime("%Y-%m-%d %H:%M:%S", gmtime(mktime(gmtime())))
AddSampleLog(Workspace=wksp, LogName='run_start', LogText=str(tmptime))
tsp_a = kernel.FloatTimeSeriesProperty("SensorA")
tsp_b = kernel.FloatTimeSeriesProperty("SensorB")
tsp_c = kernel.FloatTimeSeriesProperty("SensorC")
for i in np.arange(25):
tmptime = strftime("%Y-%m-%d %H:%M:%S",
gmtime(mktime(gmtime()) + i))
tsp_a.addValue(tmptime, 1.0 * i * i)
tsp_b.addValue(tmptime, 2.0 * i * i)
tsp_c.addValue(tmptime, 3.0 * i * i)
wksp.mutableRun()['SensorA'] = tsp_a
wksp.mutableRun()['SensorB'] = tsp_b
wksp.mutableRun()['SensorC'] = tsp_c
return wksp
def createTestWorkspace(self, run=23456):
""" Create a workspace for testing against with ideal log values
"""
from mantid.simpleapi import CreateWorkspace
from mantid.simpleapi import AddSampleLog
from time import gmtime, strftime,mktime
import numpy as np
# Create a matrix workspace
x = np.array([1.,2.,3.,4.])
y = np.array([1.,2.,3.])
e = np.sqrt(np.array([1.,2.,3.]))
wksp = CreateWorkspace(DataX=x, DataY=y,DataE=e,NSpec=1,UnitX='TOF')
# Add run_start
tmptime = strftime("%Y-%m-%d %H:%M:%S", gmtime(mktime(gmtime())))
AddSampleLog(Workspace=wksp,LogName='run_start',LogText=str(tmptime))
tsp_a=kernel.FloatTimeSeriesProperty("proton_charge")
tsp_b=kernel.FloatTimeSeriesProperty("SensorA")
for i in arange(25):
tmptime = strftime("%Y-%m-%d %H:%M:%S", gmtime(mktime(gmtime())+i))
if run == 23456:
shift = 0
else:
shift = int(run)
tsp_a.addValue(tmptime, 1.0*i*i + shift)
tsp_b.addValue(tmptime, 1.234*(i+1))
wksp.mutableRun()['run_number']=str(run)
wksp.mutableRun()['duration']=342.3
wksp.mutableRun()['SensorA'] = tsp_b
wksp.mutableRun()['proton_charge']=tsp_a
return wksp
def createTestWorkspace(self, run=23456):
""" Create a workspace for testing against with ideal log values
"""
from mantid.simpleapi import CreateWorkspace
from mantid.simpleapi import AddSampleLog
from time import gmtime, strftime,mktime
import numpy as np
# Create a matrix workspace
x = np.array([1.,2.,3.,4.])
y = np.array([1.,2.,3.])
e = np.sqrt(np.array([1.,2.,3.]))
wksp = CreateWorkspace(DataX=x, DataY=y,DataE=e,NSpec=1,UnitX='TOF')
# Add run_start
tmptime = strftime("%Y-%m-%d %H:%M:%S", gmtime(mktime(gmtime())))
AddSampleLog(Workspace=wksp,LogName='run_start',LogText=str(tmptime))
tsp_a=kernel.FloatTimeSeriesProperty("proton_charge")
tsp_b=kernel.FloatTimeSeriesProperty("SensorA")
for i in arange(25):
tmptime = strftime("%Y-%m-%d %H:%M:%S", gmtime(mktime(gmtime())+i))
if run == 23456:
shift = 0
else:
shift = int(run)
tsp_a.addValue(tmptime, 1.0*i*i + shift)
tsp_b.addValue(tmptime, 1.234*(i+1))
wksp.mutableRun()['run_number']=str(run)
wksp.mutableRun()['duration']=342.3
wksp.mutableRun()['SensorA'] = tsp_b
wksp.mutableRun()['proton_charge']=tsp_a
return wksp
def test_create_results_table_with_logs_selected(self):
workspace = CreateWorkspace([0, 1, 2, 3, 4, 5], [0, 1, 2, 3, 4, 5])
workspace.mutableRun().addProperty(
"run_start", "1970-01-01T00:00:01 to 1970-01-01T00:00:01", True)
AddTimeSeriesLog(workspace,
Name="sample_temp",
Time="2010-01-01T00:00:00",
Value=100)
AddTimeSeriesLog(workspace,
Name="sample_temp",
Time="2010-01-01T00:30:00",
Value=65)
AddTimeSeriesLog(workspace,
Name="sample_temp",
Time="2010-01-01T00:50:00",
Value=100.2)
workspace.mutableRun().addProperty("sample_magn_field", 2, True)
_, model = create_test_model(
('ws1', ), 'func1', self.parameters,
[StaticWorkspaceWrapper('ws1', workspace)], self.logs)
selected_results = [('ws1', 0)]
table = model.create_results_table(self.log_names, selected_results)
# workspace_name => no error col as its a string
# sample_temp => time series and will have non-zero error
# sample_magn_field => just a number
expected_cols = [
'workspace_name', 'run_start', 'run_start_seconds', 'sample_temp',
'sample_tempError', 'sample_magn_field', 'sample_magn_fieldError',
'f0.Height', 'f0.HeightError', 'f0.PeakCentre',
'f0.PeakCentreError', 'f0.Sigma', 'f0.SigmaError', 'f1.Height',
'f1.HeightError', 'f1.PeakCentre', 'f1.PeakCentreError',
'f1.Sigma', 'f1.SigmaError', 'Cost function value'
]
expected_types = (TableColumnType.NoType, TableColumnType.X,
TableColumnType.X, TableColumnType.X,
TableColumnType.XErr, TableColumnType.X,
TableColumnType.XErr, TableColumnType.Y,
TableColumnType.YErr, TableColumnType.Y,
TableColumnType.YErr, TableColumnType.Y,
TableColumnType.YErr, TableColumnType.Y,
TableColumnType.YErr, TableColumnType.Y,
TableColumnType.YErr, TableColumnType.Y,
TableColumnType.YErr, TableColumnType.Y)
avg_log_values = "1970-01-01T00:00:01 to 1970-01-01T00:00:01", 1, 86., 2.0
expected_content = [
('ws1_Parameters', avg_log_values[0], avg_log_values[1],
avg_log_values[2], 17.146, avg_log_values[3], 0.,
self.f0_height[0], self.f0_height[1], self.f0_centre[0],
self.f0_centre[1], self.f0_sigma[0], self.f0_sigma[1],
self.f1_height[0], self.f1_height[1], self.f1_centre[0],
self.f1_centre[1], self.f1_sigma[0], self.f1_sigma[1],
self.cost_function[0])
]
self._assert_table_matches_expected(zip(expected_cols, expected_types),
expected_content, table,
model.results_table_name())
def createTestWorkspace2(self):
""" Create a workspace for testing against with more situation
"""
from mantid.simpleapi import CreateWorkspace
from mantid.simpleapi import AddSampleLog
from time import gmtime, strftime,mktime
from datetime import datetime, timedelta
import numpy as np
# Create a matrix workspace
x = np.array([1.,2.,3.,4.])
y = np.array([1.,2.,3.])
e = np.sqrt(np.array([1.,2.,3.]))
wksp = CreateWorkspace(DataX=x, DataY=y,DataE=e,NSpec=1,UnitX='TOF')
# Add run_start
year = 2014
month = 2
day = 15
hour = 13
minute = 34
second = 3
dtimesec = 0.0010
timefluc = 0.0001
#tmptime = strftime("%Y-%m-%d %H:%M:%S", gmtime(mktime(gmtime())))
runstart = datetime(year, month, day, hour, minute, second)
AddSampleLog(Workspace=wksp,LogName='run_start',LogText=str(runstart))
tsp_a = kernel.FloatTimeSeriesProperty("SensorA")
tsp_b = kernel.FloatTimeSeriesProperty("SensorB")
tsp_c = kernel.FloatTimeSeriesProperty("SensorC")
tsp_d = kernel.FloatTimeSeriesProperty("SensorD")
logs = [tsp_a, tsp_b, tsp_c, tsp_d]
dbbuf = ""
random.seed(0)
for i in arange(25):
# Randomly pick up log without records
# first iteration must have all the record
skiploglist = []
if i > 0:
numnorecord = random.randint(-1, 4)
if numnorecord > 0:
for j in xrange(numnorecord):
logindex = random.randint(0, 6)
skiploglist.append(logindex)
# ENDFOR (j)
# ENDIF (numnorecord)
# ENDIF (i)
dbbuf += "----------- %d -------------\n" % (i)
# Record
for j in xrange(4):
# Skip if selected
if j in skiploglist:
continue
# get random time shifts
timeshift = (random.random()-0.5)*timefluc
if i == 0:
# first record should have the 'exactly' same time stamps
timeshift *= 0.0001
deltatime = timedelta(i*dtimesec + timeshift)
tmptime = str(runstart + deltatime)
tmpvalue = float(i*i*6)+j
logs[j].addValue(tmptime, tmpvalue)
dbbuf += "%s: %s = %d\n" % (logs[j].name, tmptime, tmpvalue)
# ENDFOR (j)
# ENDFOR (i)
# print dbbuf
wksp.mutableRun()['SensorA']=tsp_a
wksp.mutableRun()['SensorB']=tsp_b
wksp.mutableRun()['SensorC']=tsp_c
wksp.mutableRun()['SensorD']=tsp_d
return wksp
def createTestWorkspace2(self):
""" Create a workspace for testing against with more situation
"""
from mantid.simpleapi import CreateWorkspace
from mantid.simpleapi import AddSampleLog
import numpy
from numpy import datetime64, timedelta64
#from time import gmtime, strftime,mktime # in debug prints
# Create a matrix workspace
x = np.array([1., 2., 3., 4.])
y = np.array([1., 2., 3.])
e = np.sqrt(np.array([1., 2., 3.]))
wksp = CreateWorkspace(DataX=x, DataY=y, DataE=e, NSpec=1, UnitX='TOF')
# Add run_start
dtimesec = 0.0010
timefluc = 0.0001
runstart = '2014-02-15T13:34:03'
# older numpy assumes local timezone
if LooseVersion(numpy.__version__) < LooseVersion('1.9'):
runstart = runstart + 'Z'
runstart = datetime64(runstart, 'us') # microsecond needed for deltas
AddSampleLog(Workspace=wksp,
LogName='run_start',
LogText=str(runstart))
tsp_a = kernel.FloatTimeSeriesProperty("SensorA")
tsp_b = kernel.FloatTimeSeriesProperty("SensorB")
tsp_c = kernel.FloatTimeSeriesProperty("SensorC")
tsp_d = kernel.FloatTimeSeriesProperty("SensorD")
logs = [tsp_a, tsp_b, tsp_c, tsp_d]
dbbuf = ""
np.random.seed(0)
for i in np.arange(25):
# Randomly pick up log without records
# first iteration must have all the record
skiploglist = []
if i > 0:
numnorecord = np.random.randint(-1, 4)
if numnorecord > 0:
for j in range(numnorecord):
logindex = np.random.randint(0, 6)
skiploglist.append(logindex)
# ENDFOR (j)
# ENDIF (numnorecord)
# ENDIF (i)
dbbuf += "----------- %d -------------\n" % (i)
# Record
for j in range(4):
# Skip if selected
if j in skiploglist:
continue
# get random time shifts
timeshift = (np.random.random() - 0.5) * timefluc
if i == 0:
# first record should have the 'exactly' same time stamps
timeshift *= 0.0001
deltatime = i * dtimesec + timeshift # fraction of a day
deltatime = timedelta64(int(deltatime * 24 * 3600 * 1e6),
'us') # timedelta64 requires int
tmptime = runstart + deltatime
tmpvalue = float(i * i * 6) + j
logs[j].addValue(tmptime, tmpvalue)
#dbbuf += "{}: {} = {}\n".format(logs[j].name, tmptime, tmpvalue)
# ENDFOR (j)
# ENDFOR (i)
#print(dbbuf)
wksp.mutableRun()['SensorA'] = tsp_a
wksp.mutableRun()['SensorB'] = tsp_b
wksp.mutableRun()['SensorC'] = tsp_c
wksp.mutableRun()['SensorD'] = tsp_d
return wksp
def createTestWorkspace2(self):
""" Create a workspace for testing against with more situation
"""
from mantid.simpleapi import CreateWorkspace
from mantid.simpleapi import AddSampleLog
import numpy
from numpy import datetime64, timedelta64
#from time import gmtime, strftime,mktime # in debug prints
# Create a matrix workspace
x = np.array([1.,2.,3.,4.])
y = np.array([1.,2.,3.])
e = np.sqrt(np.array([1.,2.,3.]))
wksp = CreateWorkspace(DataX=x, DataY=y,DataE=e,NSpec=1,UnitX='TOF')
# Add run_start
dtimesec = 0.0010
timefluc = 0.0001
runstart = '2014-02-15T13:34:03'
# older numpy assumes local timezone
if LooseVersion(numpy.__version__) < LooseVersion('1.9'):
runstart = runstart + 'Z'
runstart = datetime64(runstart, 'us') # microsecond needed for deltas
AddSampleLog(Workspace=wksp,LogName='run_start',LogText=str(runstart))
tsp_a = kernel.FloatTimeSeriesProperty("SensorA")
tsp_b = kernel.FloatTimeSeriesProperty("SensorB")
tsp_c = kernel.FloatTimeSeriesProperty("SensorC")
tsp_d = kernel.FloatTimeSeriesProperty("SensorD")
logs = [tsp_a, tsp_b, tsp_c, tsp_d]
dbbuf = ""
np.random.seed(0)
for i in np.arange(25):
# Randomly pick up log without records
# first iteration must have all the record
skiploglist = []
if i > 0:
numnorecord = np.random.randint(-1, 4)
if numnorecord > 0:
for j in range(numnorecord):
logindex = np.random.randint(0, 6)
skiploglist.append(logindex)
# ENDFOR (j)
# ENDIF (numnorecord)
# ENDIF (i)
dbbuf += "----------- %d -------------\n" % (i)
# Record
for j in range(4):
# Skip if selected
if j in skiploglist:
continue
# get random time shifts
timeshift = (np.random.random()-0.5)*timefluc
if i == 0:
# first record should have the 'exactly' same time stamps
timeshift *= 0.0001
deltatime = i*dtimesec + timeshift # fraction of a day
deltatime = timedelta64(int(deltatime * 24 * 3600 * 1e6), 'us') # timedelta64 requires int
tmptime = runstart + deltatime
tmpvalue = float(i*i*6)+j
logs[j].addValue(tmptime, tmpvalue)
#dbbuf += "{}: {} = {}\n".format(logs[j].name, tmptime, tmpvalue)
# ENDFOR (j)
# ENDFOR (i)
#print(dbbuf)
wksp.mutableRun()['SensorA']=tsp_a
wksp.mutableRun()['SensorB']=tsp_b
wksp.mutableRun()['SensorC']=tsp_c
wksp.mutableRun()['SensorD']=tsp_d
return wksp
def createTestWorkspace2(self):
""" Create a workspace for testing against with more situation
"""
from mantid.simpleapi import CreateWorkspace
from mantid.simpleapi import AddSampleLog
from time import gmtime, strftime, mktime
from datetime import datetime, timedelta
import numpy as np
# Create a matrix workspace
x = np.array([1., 2., 3., 4.])
y = np.array([1., 2., 3.])
e = np.sqrt(np.array([1., 2., 3.]))
wksp = CreateWorkspace(DataX=x, DataY=y, DataE=e, NSpec=1, UnitX='TOF')
# Add run_start
year = 2014
month = 2
day = 15
hour = 13
minute = 34
second = 3
dtimesec = 0.0010
timefluc = 0.0001
#tmptime = strftime("%Y-%m-%d %H:%M:%S", gmtime(mktime(gmtime())))
runstart = datetime(year, month, day, hour, minute, second)
AddSampleLog(Workspace=wksp,
LogName='run_start',
LogText=str(runstart))
tsp_a = kernel.FloatTimeSeriesProperty("SensorA")
tsp_b = kernel.FloatTimeSeriesProperty("SensorB")
tsp_c = kernel.FloatTimeSeriesProperty("SensorC")
tsp_d = kernel.FloatTimeSeriesProperty("SensorD")
logs = [tsp_a, tsp_b, tsp_c, tsp_d]
dbbuf = ""
random.seed(0)
for i in arange(25):
# Randomly pick up log without records
# first iteration must have all the record
skiploglist = []
if i > 0:
numnorecord = random.randint(-1, 4)
if numnorecord > 0:
for j in range(numnorecord):
logindex = random.randint(0, 6)
skiploglist.append(logindex)
# ENDFOR (j)
# ENDIF (numnorecord)
# ENDIF (i)
dbbuf += "----------- %d -------------\n" % (i)
# Record
for j in range(4):
# Skip if selected
if j in skiploglist:
continue
# get random time shifts
timeshift = (random.random() - 0.5) * timefluc
if i == 0:
# first record should have the 'exactly' same time stamps
timeshift *= 0.0001
deltatime = timedelta(i * dtimesec + timeshift)
tmptime = str(runstart + deltatime)
tmpvalue = float(i * i * 6) + j
logs[j].addValue(tmptime, tmpvalue)
dbbuf += "%s: %s = %d\n" % (logs[j].name, tmptime, tmpvalue)
# ENDFOR (j)
# ENDFOR (i)
# print dbbuf
wksp.mutableRun()['SensorA'] = tsp_a
wksp.mutableRun()['SensorB'] = tsp_b
wksp.mutableRun()['SensorC'] = tsp_c
wksp.mutableRun()['SensorD'] = tsp_d
return wksp
