q1_2.py

test = {   'name': 'q1_2',
    'points': None,
    'suites': [   {   'cases': [   {'code': '>>> plot_type in {1, 2, 3, 4} # your answer should be one of these\nTrue', 'hidden': False, 'locked': False},
                                   {'code': '>>> plot_type == 1\nTrue', 'hidden': False, 'locked': False}],
                      'scored': True,
                      'setup': '',
                      'teardown': '',
                      'type': 'doctest'}]}