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)