mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-28 02:02:39 +08:00
Lindsey Heagy
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import unittest
|
|
import sys
|
|
import os
|
|
from SimPEG import Examples
|
|
import numpy as np
|
|
|
|
class compareInitFiles(unittest.TestCase):
|
|
def test_compareInitFiles(self):
|
|
print 'Checking that __init__.py up-to-date in SimPEG/Examples'
|
|
fName = os.path.abspath(__file__)
|
|
ExamplesDir = os.path.sep.join(fName.split(os.path.sep)[:-3] + ['SimPEG', 'Examples'])
|
|
|
|
files = os.listdir(ExamplesDir)
|
|
|
|
pyfiles = []
|
|
[pyfiles.append(py.rstrip('.py')) for py in files if py.endswith('.py') and py != '__init__.py']
|
|
|
|
setdiff = set(pyfiles) - set(Examples.__examples__)
|
|
|
|
print ' Any missing files? ', setdiff
|
|
|
|
didpass = (setdiff == set())
|
|
|
|
self.assertTrue(didpass, "Examples not up to date, run 'python __init__.py' from SimPEG/Examples to update")
|
|
|
|
def get(test):
|
|
def test_func(self):
|
|
print '\nTesting %s.run(plotIt=False)\n'%test
|
|
getattr(Examples, test).run(plotIt=False)
|
|
self.assertTrue(True)
|
|
return test_func
|
|
attrs = dict()
|
|
|
|
for test in Examples.__examples__:
|
|
attrs['test_'+test] = get(test)
|
|
|
|
TestExamples = type('TestExamples', (unittest.TestCase,), attrs)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|