added gradient

1 files changed, 8 insertions, 0 deletions

diff --git a/Pb/__init__.py b/Pb/__init__.py
index 6485e86..af4558f 100644
--- a/Pb/__init__.py
+++ b/Pb/__init__.py
@@ -106,4 +106,12 @@ class Pb(object):
   def err_fatal(self, s):
     sys.stderr.write("ERROR[FATAL]:{} - {}\n".format(self.__class__.__name__, s))
     sys.exit(1);
+  
+  @classmethod
+  def test(cls, test_params=None):
+    test_params = test_params or cls._test_params
+    if not test_params:
+      raise AttributeError ("Must supply test params to test %s" % cls.__name__)
+    b = cls(**test_params)
+    b.create();