wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-08-07 11:28:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

light pep8 formatting in tests

Browse Source
This commit is contained in:
Lindsey Heagy
2016-07-16 16:26:36 -07:00
parent 6826df8989
commit 62b3a3a218
3 changed files with 18 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/base/test_problem.py
+3 -3
View File
@@ -5,12 +5,12 @@ from SimPEG import *
class TestTimeProblem(unittest.TestCase):
def setUp(self):
mesh = Mesh.TensorMesh([10,10])
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]
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
@@ -18,7 +18,7 @@ class TestTimeProblem(unittest.TestCase):
self.assertTrue(self.prob.nT == 6)
self.assertTrue(np.all(self.prob.times == np.r_[0,trueTS].cumsum()))
self.assertTrue(np.all(self.prob.times == np.r_[0, trueTS].cumsum()))
if __name__ == '__main__':