Merge pull request #3520 from julienr/optdb_contains

theano/gof/optdb.py

@@ -160,6 +160,9 @@ multiple time in a DB. Tryed to register "%s" again under the new name "%s".
         for variable in variables:
             return variable
 
+    def __contains__(self, name):
+        return name in self.__db__
+
     def print_summary(self, stream=sys.stdout):
         print("%s (id %i)" % (self.__class__.__name__, id(self)), file=stream)
         print("  names", self._names, file=stream)

theano/gof/tests/test_optdb.py

@@ -18,6 +18,10 @@ class Test_DB(TestCase):
 
         db.register('c', Opt(), 'z', 'asdf')
 
+        self.assertTrue('a' in db)
+        self.assertTrue('b' in db)
+        self.assertTrue('c' in db)
+
         try:
             db.register('c', Opt())  # name taken
             self.fail()