Spaces:
Running
Running
File size: 676 Bytes
dd39c08 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import pytest
from browsergym.core.action.python import PythonActionSet
ACTIONS_TO_TEST = [
(
"""\
a = 0
""",
"""\
a = 0
""",
),
(
"""\
```
a = 0
```
""",
"""\
a = 0
""",
),
(
"""\
```python
a = 0
```
""",
"""\
a = 0
""",
),
(
"""\
```python
a = 0
```
This is an explanation
```python
b = 3
```
More explanations
""",
"""\
a = 0
b = 3
""",
),
]
@pytest.mark.parametrize("action,expected_code", ACTIONS_TO_TEST)
def test_action_cleaning(action, expected_code):
action_set = PythonActionSet()
code = action_set.to_python_code(action)
assert code == expected_code
|