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)