def METHOD_NAME(self):
    ts = test_set.BanditTestSet(self.config)
    self.assertEqual(1, len(ts.get_tests("Import")))
    self.assertEqual(1, len(ts.get_tests("ImportFrom")))
    self.assertEqual(1, len(ts.get_tests("Call")))