def test_warn_invalid_flags(self):
with set_refprune_flags('abc,per_bb,cde'):
with self.assertWarns(NumbaInvalidConfigWarning) as cm:
optval = _parse_refprune_flags()
self.assertEqual(len(cm.warnings), 2)
self.assertIn('abc', str(cm.warnings[0].message))
self.assertIn('cde', str(cm.warnings[1].message))
self.assertEqual(optval, llvm.RefPruneSubpasses.PER_BB)
def test_some_flags(self):
with set_refprune_flags('per_bb, fanout'):
optval = _parse_refprune_flags()
enumcls = llvm.RefPruneSubpasses
self.assertEqual(optval, enumcls.PER_BB | enumcls.FANOUT)
def test_the_all_flag(self):
with set_refprune_flags('all'):
optval = _parse_refprune_flags()
self.assertEqual(optval, llvm.RefPruneSubpasses.ALL)
def test_valid_flag(self):
with set_refprune_flags('per_bb, diamond, fanout,fanout_raise'):
optval = _parse_refprune_flags()
self.assertEqual(optval, llvm.RefPruneSubpasses.ALL)
