he-017

# HumanEval/17

Implement a file **`solution.py`** that completes the function below. Keep the given name and signature; your file must define it at module level.

```python
from typing import List


def parse_music(music_string: str) -> List[int]:
    """ Input to this function is a string representing musical notes in a special ASCII format.
    Your task is to parse this string and return list of integers corresponding to how many beats does each
    not last.

    Here is a legend:
    'o' - whole note, lasts four beats
    'o|' - half note, lasts two beats
    '.|' - quater note, lasts one beat

    >>> parse_music('o o| .| o| o| .| .| .| .| o o')
    [4, 2, 1, 2, 2, 1, 1, 1, 1, 4, 4]
    """
```

<!-- imported from OpenAI HumanEval (HumanEval/17) -->

# Auto-generated from OpenAI HumanEval HumanEval/17. Do not edit by hand.
from solution import *  # noqa: F401,F403 (prompt helpers may be referenced by tests)
from solution import parse_music as candidate

METADATA = {
    'author': 'jt',
    'dataset': 'test'
}


def check(candidate):
    assert candidate('') == []
    assert candidate('o o o o') == [4, 4, 4, 4]
    assert candidate('.| .| .| .|') == [1, 1, 1, 1]
    assert candidate('o| o| .| .| o o o o') == [2, 2, 1, 1, 4, 4, 4, 4]
    assert candidate('o| .| o| .| o o| o o|') == [2, 1, 2, 1, 4, 2, 4, 2]


def test_he_017():
    check(candidate)

You are an expert programmer. Solve the task exactly as specified. Output your solution as fenced code blocks using the required file name(s) and the exact function/type signatures requested. Prefer correctness; do not include prose outside code unless asked.

```python
from typing import List

def parse_music(music_string: str) -> List[int]:
    note_map = {
        'o': 4,
        'o|': 2,
        '.|': 1
    }
    result = []
    i = 0
    n = len(music_string)
    while i < n:
        if music_string[i] == 'o':
            result.append(4)
            i += 1
        elif music_string[i] == 'o' and i + 1 < n and music_string[i+1] == '|':
            result.append(2)
            i += 2
        elif music_string[i] == '.' and i + 1 < n and music_string[i+1] == '|':
            result.append(1)
            i += 2
        else:
            i += 1
    return result
```

[
  {
    "total": 1,
    "answer": "```python\nfrom typing import List\n\ndef parse_music(music_string: str) -> List[int]:\n    note_map = {\n        'o': 4,\n        'o|': 2,\n        '.|': 1\n    }\n    result = []\n    i = 0\n    n = len(music_string)\n    while i < n:\n        if music_string[i] == 'o':\n            result.append(4)\n            i += 1\n        elif music_string[i] == 'o' and i + 1 < n and music_string[i+1] == '|':\n            result.append(2)\n            i += 2\n        elif music_string[i] == '.' and i + 1 < n and music_string[i+1] == '|':\n            result.append(1)\n            i += 2\n        else:\n            i += 1\n    return result\n```",
    "passed": 0,
    "test_error": "F                                                                        [100%]\n=================================== FAILURES ===================================\n_________________________________ test_he_017 __________________________________\n\n    def test_he_017():\n>       check(candidate)\n\ntest_he_017.py:20: \n_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ \n\ncandidate = <function parse_music at 0x7ee9e01b47c0>\n\n    def check(candidate):\n        assert candidate('') == []\n        assert candidate('o o o o') == [4, 4, 4, 4]\n        assert candidate('.| .| .| .|') == [1, 1, 1, 1]\n>       assert candidate('o| o| .| .| o o o o') == [2, 2, 1, 1, 4, 4, 4, 4]\nE       assert [4, 4, 1, 1, 4, 4, ...] == [2, 2, 1, 1, 4, 4, ...]\nE         \nE         At index 0 diff: 4 != 2\nE         Use -v to get more diff\n\ntest_he_017.py:15: AssertionError\n=========================== short test summary info ============================\nFAILED test_he_017.py::test_he_017 - assert [4, 4, 1, 1, 4, 4, ...] == [2, 2,...\n1 failed in 0.02s"
  },
  {
    "total": 1,
    "answer": "```python\nfrom typing import List\n\ndef parse_music(music_string: str) -> List[int]:\n    note_map = {\n        'o': 4,\n        'o|': 2,\n        '.|': 1\n    }\n    result = []\n    tokens = music_string.split()\n    for token in tokens:\n        if token in note_map:\n            result.append(note_map[token])\n    return result\n```",
    "passed": 1,
    "test_error": ""
  }
]