(sct, stu_code, passes, msg)
| 122 | ], |
| 123 | ) |
| 124 | def test_test_data_frame(sct, stu_code, passes, msg): |
| 125 | output = helper.run( |
| 126 | { |
| 127 | "DC_PEC": "import pandas as pd", |
| 128 | "DC_SOLUTION": 'df = pd.DataFrame({"a": [1, 2, 3]})', |
| 129 | "DC_CODE": stu_code, |
| 130 | "DC_SCT": sct, |
| 131 | } |
| 132 | ) |
| 133 | assert output["correct"] == passes |
| 134 | if msg: |
| 135 | assert output["message"] == msg |
| 136 | |
| 137 | |
| 138 | @pytest.mark.parametrize( |