def testParticularSpectra():
spectra = kiknet.spectraForGmNos([2] , ['pga', 0.02, 0.03] , ['B1'])
#print spectra['B1']
assert spectra['B1'] == [[2, 2, 0.00038584, 0.00039089, 0.00040397]]
spectra = kiknet.spectraForGmNos([2, 5] , ['pga', 0.02, 0.03] , ['B1'])
assert spectra['B1'] == [[2, 2, 0.00038584, 0.00039089, 0.00040397] , [5, 7.7778, 0.0021254, 0.0021487, 0.002182]]
def checkSpectraForGmNos(gmNos, periods, components):
data = kiknet.spectraForGmNos(gmNos, periods, components)
# Check that data is returned
dataPresent = [len(data[d]) > 0 for d in data]
assert all(dataPresent)
# check that name of dict keys are same as component list
if components == []:
components = ['MS']
assert data.keys().sort() == components.sort()
# check that the number of periods returned is same as requested
if periods == []:
periods = [0]*24
if periods[0] == -1:
periods = [0]*105
sameNumberOfPeriods = [len(s)-2 == len(periods) for d in data for s in data[d]]
assert all(sameNumberOfPeriods)
# check that the same gmNos are returnes
returnedGmNos = [[s[0] for s in data[d]] for d in data]
gmNos.sort()
sameGmNos = [gm == gmNos for gm in returnedGmNos]
assert all(sameGmNos)
