def testExpected_maxTimeSet(self):
hillA, hillB, hillC = self._loadMultiDimHillsInTimeOrder()
self.timeRange = [0, 2.2]
groupA = tCode.GroupedMultiDimGaussHills([hillA])
groupB = tCode.GroupedMultiDimGaussHills([hillA, hillB])
expTimesVsHills = [[1, groupA], [2, groupB]]
actTimesVsHills = self._runTestFunct()
self.assertEqual(expTimesVsHills, actTimesVsHills)
def testExpectedFunctValA_fullRange(self):
#Figure out the expected potential objects
hillA, hillB, hillC = self._loadMultiDimHillsInTimeOrder()
groupA = tCode.GroupedMultiDimGaussHills([hillA])
groupB = tCode.GroupedMultiDimGaussHills([hillA, hillB])
groupC = tCode.GroupedMultiDimGaussHills([hillA, hillB, hillC])
expTimesVsHills = [[1, groupA], [2, groupB], [3, groupC]]
actTimesVsHills = self._runTestFunct()
self.assertEqual(expTimesVsHills, actTimesVsHills)
def testExpected_ridicLargeMinTimeDiff(self):
self.minTimeDiff = 10
hillA, hillB, hillC = self._loadMultiDimHillsInTimeOrder()
groupA = tCode.GroupedMultiDimGaussHills([hillA, hillB, hillC])
expTimesVsHills = [[1, groupA]]
actTimesVsHills = self._runTestFunct()
self.assertEqual(expTimesVsHills, actTimesVsHills)
def _loadExpectedCaseA(self):
multiDimGaus = list()
for idx in range(3):
currMultiDimGau = metadynHillsHelp.MultiDimGaussHill.fromIters(
heights=[self.heightsA[idx], self.heightsB[idx]],
scales=[self.scalesA[idx], self.scalesB[idx]],
positions=[self.positionsA[idx], self.positionsB[idx]])
multiDimGaus.append(currMultiDimGau)
expObj = metadynHillsHelp.GroupedMultiDimGaussHills(multiDimGaus)
return expObj
def createTestObjs(self):
self.testObjA = tCode.GroupedMultiDimGaussHills([self.gauA, self.gauB])