mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-28 02:02:39 +08:00
Lindsey Heagy
26 lines
694 B
Python
26 lines
694 B
Python
import unittest
|
|
from SimPEG import *
|
|
|
|
|
|
class TestTimeProblem(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
mesh = Mesh.TensorMesh([10, 10])
|
|
self.prob = Problem.BaseTimeProblem(mesh)
|
|
|
|
def test_timeProblem_setTimeSteps(self):
|
|
self.prob.timeSteps = [(1e-6, 3), 1e-5, (1e-4, 2)]
|
|
trueTS = np.r_[1e-6, 1e-6, 1e-6, 1e-5, 1e-4, 1e-4]
|
|
self.assertTrue(np.all(trueTS == self.prob.timeSteps))
|
|
|
|
self.prob.timeSteps = trueTS
|
|
self.assertTrue(np.all(trueTS == self.prob.timeSteps))
|
|
|
|
self.assertTrue(self.prob.nT == 6)
|
|
|
|
self.assertTrue(np.all(self.prob.times == np.r_[0, trueTS].cumsum()))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|