Tests.py

import unittest
from RegExpBuilder import RegExpBuilder

class Test(unittest.TestCase):
    def test_start(self):
        regex = RegExpBuilder()
        regex.start()
        regex.exactly(1).of("p")
        regex = regex.getRegExp()
    
        self.assertTrue(regex.match("p") is not None)
        self.assertTrue(regex.match("qp") is None)
  
    def test_end(self):
        regex = RegExpBuilder()
        regex.exactly(1).of("p")
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("p") is not None)
        self.assertTrue(regex.match("pq") is None)
  
    def test_eitherLike_orLike(self):
        regex = RegExpBuilder()
        regex.start()
        regex.eitherLike(lambda r: r.exactly(1).of("p"))
        regex.orLike(lambda r: r.exactly(2).of("q"))
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("p") is not None)
        self.assertTrue(regex.match("qq") is not None)
        self.assertTrue(regex.match("pqq") is None)
        self.assertTrue(regex.match("qqp") is None)
  
    def test_exactly(self):
        regex = RegExpBuilder()
        regex.start()
        regex.exactly(3).of("p")
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("ppp") is not None)
        self.assertTrue(regex.match("pp") is None)
        self.assertTrue(regex.match("pppp") is None) 
  
    def test_min(self):
        regex = RegExpBuilder()
        regex.start()
        regex.min(2).of("p")
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("pp") is not None)
        self.assertTrue(regex.match("ppp") is not None)
        self.assertTrue(regex.match("ppppppp") is not None)
        self.assertTrue(regex.match("p") is None)
  
    def test_max(self):
        regex = RegExpBuilder()
        regex.start()
        regex.max(3).of("p")
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("p") is not None)
        self.assertTrue(regex.match("pp") is not None)
        self.assertTrue(regex.match("ppp") is not None)
        self.assertTrue(regex.match("pppp") is None)
        self.assertTrue(regex.match("pppppppp") is None)
  
    def test_min_max(self):
        regex = RegExpBuilder()
        regex.start()
        regex.min(3).max(7).of("p")
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("ppp") is not None)
        self.assertTrue(regex.match("ppppp") is not None)
        self.assertTrue(regex.match("ppppppp") is not None)
        self.assertTrue(regex.match("pp") is None)
        self.assertTrue(regex.match("p") is None)
        self.assertTrue(regex.match("pppppppp") is None)
        self.assertTrue(regex.match("pppppppppppp") is None)
  
    def test_of(self):
        regex = RegExpBuilder()
        regex.start()
        regex.exactly(2).of("p p p ")
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("p p p p p p ") is not None)
        self.assertTrue(regex.match("p p p p pp") is None)
  
    def test_ofAny(self):
        regex = RegExpBuilder()
        regex.start()
        regex.exactly(3).ofAny()
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("pqr") is not None)
  
    def test_fromClass(self):
        someLetters = ["p", "q", "r"]
        regex = RegExpBuilder()
        regex.start()
        regex.exactly(3).fromClass(someLetters)
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("ppp") is not None)
        self.assertTrue(regex.match("qqq") is not None)
        self.assertTrue(regex.match("ppq") is not None)
        self.assertTrue(regex.match("rqp") is not None)
        self.assertTrue(regex.match("pyy") is None)
  
    def test_notFromClass(self):
        someLetters = ["p", "q", "r"]
        regex = RegExpBuilder()
        regex.start()
        regex.exactly(3).notFromClass(someLetters)
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("lmn") is not None)
        self.assertTrue(regex.match("mnq") is None)
  
    def test_like(self):
        pattern = lambda r: r.min(1).of("p").min(2).of("q")
        
        regex = RegExpBuilder()
        regex.start()
        regex.exactly(2).like(pattern)
        regex.end()
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("pqqpqq") is not None)
        self.assertTrue(regex.match("qppqpp") is None)
  
    def test_reluctantly(self):
        regex = RegExpBuilder()
        regex.exactly(2).of("p")
        regex.min(2).ofAny().reluctantly()
        regex.exactly(2).of("p")
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("pprrrrpprrpp").group() == "pprrrrpp")
  
    def test_behind(self):
        regex = RegExpBuilder()
        regex.exactly(1).of("dart")
        regex.behind(lambda r: r.exactly(1).of("lang"))
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("dartlang").group() == "dart")
        self.assertTrue(regex.match("dartpqr") is None)
  
    def test_notBehind(self):
        regex = RegExpBuilder()
        regex.exactly(1).of("dart")
        regex.notBehind(lambda r: r.exactly(1).of("pqr"))
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("dartlang") is not None)
        self.assertTrue(regex.match("dartpqr") is None)
  
    def test_asCapturingGroup(self):
        regex = RegExpBuilder()
        regex.min(1).max(3).of("p")
        regex.exactly(1).of("dart").asCapturingGroup()
        regex.exactly(1).fromClass(["p", "q", "r"])
        regex = regex.getRegExp()
        
        self.assertTrue(regex.match("pdartq").group(1) == "dart")

if __name__ == '__main__':
    unittest.main()