diff --git a/mir_eval/chord.py b/mir_eval/chord.py index a5b2d64d..7cf0ba67 100644 --- a/mir_eval/chord.py +++ b/mir_eval/chord.py @@ -258,6 +258,7 @@ def scale_degree_to_bitmap(scale_degree, modulo=False, length=BITMAP_LENGTH): "min6": [1, 0, 0, 1, 0, 0, 0, 1, 0, 1, 0, 0], "dim7": [1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0], "hdim7": [1, 0, 0, 1, 0, 0, 1, 0, 0, 0, 1, 0], + "aug7": [1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 1, 0], "maj9": [1, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 1], "min9": [1, 0, 0, 1, 0, 0, 0, 1, 0, 0, 1, 0], "9": [1, 0, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0], diff --git a/tests/test_chord.py b/tests/test_chord.py index e8db17fa..839b182f 100644 --- a/tests/test_chord.py +++ b/tests/test_chord.py @@ -180,6 +180,7 @@ def test_rotate_bitmaps_to_roots(bitmap, root, expected_bitmap): ("A:9/b3", 9, [1, 0, 0, 1, 1, 0, 0, 1, 0, 0, 1, 0], 3, False, False), ("G:dim(4)/6", 7, [1, 0, 0, 1, 0, 1, 1, 0, 0, 1, 0, 0], 9, False, False), ("A:9", 9, [1, 0, 1, 0, 1, 0, 0, 1, 0, 0, 1, 0], 0, True, False), + ("Ab:aug7", 8, [1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 1, 0], 0, True, False), ], ) def test_chord_encode(label, e_root, e_interval, e_bass, reduce, strict):