def test_five_args(self):
resp = _extract_args("a1, a2=3, a3, a4='watermellon', a5=b37:")
self.assertSetEqual({'a1', 'a2', 'a3', 'a4', 'a5'}, resp)
def test_collection_args(self):
resp = _extract_args(
'x1=[a, b, c], x2 = {5: "32", 4:[21,22,23]}, x3 = (x+22):')
self.assertSetEqual({'x1', 'x2', 'x3'}, resp)
def test_two_long_args(self):
resp = _extract_args(" long_var546 , longer_var5932jke : ")
self.assertSetEqual({"long_var546", "longer_var5932jke"}, resp)
def test_two_default_arg(self):
resp = _extract_args("x,opt=5")
self.assertSetEqual({"x", "opt"}, resp)
def test_default_arg(self):
resp = _extract_args("x=5")
self.assertSetEqual({"x"}, resp)
def test_two_args(self):
resp = _extract_args("x, y:")
self.assertSetEqual({"x", "y"}, resp)
def test_single_arg(self):
resp = _extract_args("x:")
self.assertSetEqual({"x"}, resp)