def test_bad_anchoring(self):
with self.assertRaisesRegexp(ValueError, 'anchoring must be one of.*'):
re2.Set(None)
with self.assertRaisesRegexp(ValueError, 'anchoring must be one of.*'):
re2.Set(15)
with self.assertRaisesRegexp(ValueError, 'anchoring must be one of.*'):
re2.Set({})
def test_set_unanchored(self):
s = re2.Set(re2.UNANCHORED)
s_with_default_anchoring = re2.Set()
for re2_set in [s, s_with_default_anchoring]:
re2_set.add('foo')
re2_set.add('bar')
re2_set.add('baz')
re2_set.compile()
self.assertEqual(re2_set.match('foo'), [0])
self.assertEqual(re2_set.match('bar'), [1])
self.assertEqual(re2_set.match('baz'), [2])
self.assertEqual(re2_set.match('afoobaryo'), [0, 1])
self.assertEqual(re2_set.match('ooba'), [])
def test_add_with_bad_pattern(self):
s = re2.Set()
with self.assertRaisesRegexp(ValueError, 'missing \)'):
s.add('(')
with self.assertRaises(TypeError):
s.add(3)
def test_add_after_compile(self):
s = re2.Set()
s.add('foo')
s.compile()
with self.assertRaisesRegexp(
RuntimeError, 'Can\'t add\(\) on an already compiled Set'):
s.add('bar')
def test_set_anchor_start(self):
s = re2.Set(re2.ANCHOR_START)
self.assertEqual(s.add('foo'), 0)
self.assertEqual(s.add('bar'), 1)
s.compile()
self.assertEqual(s.match('foobar'), [0])
self.assertEqual(s.match('oobar'), [])
self.assertEqual(s.match('foo'), [0])
self.assertEqual(s.match('ofoobaro'), [])
self.assertEqual(s.match('baro'), [1])
def test_set_anchor_both(self):
s = re2.Set(re2.ANCHOR_BOTH)
self.assertEqual(s.add('foo'), 0)
self.assertEqual(s.add('bar'), 1)
s.compile()
self.assertEqual(s.match('foobar'), [])
self.assertEqual(s.match('fooba'), [])
self.assertEqual(s.match('oobar'), [])
self.assertEqual(s.match('foo'), [0])
self.assertEqual(s.match('bar'), [1])
def test_double_compile(self):
s = re2.Set()
s.add('foo')
s.compile()
s.compile()
def test_match_without_compile(self):
s = re2.Set()
s.add('foo')
with self.assertRaisesRegexp(RuntimeError, 'Can\'t match\(\) on an.*'):
s.match('bar')
