Key word args for the model.test() function that go to checkDeriv

This commit is contained in:
rowanc1
2014-03-23 10:22:39 -07:00
parent 61a08408d0
commit a3e7b96d2e
+4 -2
View File
@@ -58,11 +58,13 @@ class BaseModel(object):
def example(self):
return np.random.rand(self.nP)
def test(self, m=None):
def test(self, m=None, **kwargs):
print 'Testing the %s Class!' % self.__class__.__name__
if m is None:
m = self.example()
return checkDerivative(lambda m : [self.transform(m), self.transformDeriv(m)], m, plotIt=False)
if 'plotIt' not in kwargs:
kwargs['plotIt'] = False
return checkDerivative(lambda m : [self.transform(m), self.transformDeriv(m)], m, **kwargs)
class BaseNonLinearModel(object):
"""