def test_get_count_bad_table(self): with self.assertRaises(ValueError): get_count('BADTABLE')
def test_get_count(self): obs = get_count('project') self.assertEqual(obs, 3)