pyrekordbox 0.3.2__py3-none-any.whl → 0.4.0__py3-none-any.whl

tests/test_anlz.py

@@ -1,206 +0,0 @@
-# -*- coding: utf-8 -*-
-# Author: Dylan Jones
-# Date:   2023-02-01
-
-import os
-import pytest
-import numpy as np
-from numpy.testing import assert_equal
-from pyrekordbox import anlz
-
-TEST_ROOT = os.path.join(os.path.dirname(os.path.dirname(__file__)), ".testdata")
-ANLZ_ROOT = os.path.join(TEST_ROOT, "export", "PIONEER", "USBANLZ")
-ANLZ_DIRS = list(anlz.walk_anlz_paths(ANLZ_ROOT))
-ANLZ_FILES = [paths for _, paths in ANLZ_DIRS]
-
-
-def test_parse():
-    for root, files in ANLZ_DIRS:
-        for path in files.values():
-            anlz.AnlzFile.parse_file(path)
-
-
-def test_rebuild():
-    for root, files in ANLZ_DIRS:
-        for path in files.values():
-            file = anlz.AnlzFile.parse_file(path)
-            data = file.build()
-            assert len(data) == file.file_header.len_file
-            _ = anlz.AnlzFile.parse(data)
-
-
-def test_read_anlz_files():
-    for root, files in ANLZ_DIRS:
-        anlz_files = anlz.read_anlz_files(root)
-        assert len(files) == len(anlz_files)
-
-
-# -- Tags ------------------------------------------------------------------------------
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pqtz_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-    beats, bpms, times = tag.get()
-    nbeats = len(beats)
-    # Check that shape of arrays are equal
-    assert nbeats == len(bpms)
-    assert nbeats == len(times)
-    # Check that the beats array only contains the values 1-4
-    if nbeats:
-        assert_equal(np.sort(np.unique(beats)), [1, 2, 3, 4])
-
-    # Check other getters
-    assert_equal(beats, tag.get_beats())
-    assert_equal(bpms, tag.get_bpms())
-    assert_equal(times, tag.get_times())
-
-
-def test_pqtz_tag_set_beats():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-
-    beats = np.ones(tag.count)
-    tag.set_beats(beats)
-    assert_equal(tag.get_beats(), beats)
-
-
-def test_pqtz_tag_set_bpms():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-
-    bpms = 100 * np.ones(tag.count, dtype=np.float64)
-    tag.set_bpms(bpms)
-    assert_equal(tag.get_bpms(), bpms)
-
-
-def test_pqtz_tag_set_times():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-
-    times = 0.5 * np.arange(tag.count)
-    tag.set_times(times)
-    assert_equal(tag.get_times(), times)
-
-
-def test_pqtz_tag_set():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PQTZ")
-
-    beats = np.ones(tag.count)
-    bpms = 100 * np.ones(tag.count, dtype=np.float64)
-    times = 0.5 * np.arange(tag.count)
-    tag.set(beats, bpms, times)
-    assert_equal(tag.get_bpms(), bpms)
-    assert_equal(tag.get_bpms(), bpms)
-    assert_equal(tag.get_times(), times)
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_ppth_tag_getters(paths):
-    tmp = ""
-    for path in paths.values():
-        file = anlz.AnlzFile.parse_file(path)
-        tag = file.get_tag("PPTH")
-        p = tag.get()
-        if not tmp:
-            tmp = p
-        else:
-            assert tmp == p
-
-
-def test_ppth_tag_setters():
-    paths = ANLZ_FILES[0]
-    extected = r"C:/new/path/to/file.mp3"
-    for path in paths.values():
-        file = anlz.AnlzFile.parse_file(path)
-        tag = file.get_tag("PPTH")
-        tag.set(r"C:\new\path\to\file.mp3")
-        assert tag.get() == extected
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwav_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PWAV")
-
-    heights, color = tag.get()
-    assert len(heights) == len(color)
-    assert np.all(np.logical_and(0 <= heights, heights <= 31))
-    assert np.all(np.logical_and(0 <= color, color <= 7))
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwv2_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    tag = file.get_tag("PWV2")
-
-    heights, color = tag.get()
-    assert len(heights) == len(color)
-    assert np.all(np.logical_and(0 <= heights, heights <= 31))
-    assert np.all(np.logical_and(0 <= color, color <= 7))
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwv3_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["EXT"])
-    tag = file.get_tag("PWV3")
-
-    heights, color = tag.get()
-    assert len(heights) == len(color)
-    assert np.all(np.logical_and(0 <= heights, heights <= 31))
-    assert np.all(np.logical_and(0 <= color, color <= 7))
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwv4_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["EXT"])
-    tag = file.get_tag("PWV4")
-
-    heights, colors, blues = tag.get()
-    assert len(heights) == len(colors)
-    assert len(heights) == len(blues)
-
-
-@pytest.mark.parametrize("paths", ANLZ_FILES)
-def test_pwv5_tag_getters(paths):
-    file = anlz.AnlzFile.parse_file(paths["EXT"])
-    tag = file.get_tag("PWV5")
-
-    heights, colors = tag.get()
-    assert len(heights) == colors.shape[0]
-
-
-# -- File ------------------------------------------------------------------------------
-
-
-def test_anlzfile_getall_tags():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    key = "PPTH"
-    tags = file.getall_tags(key)
-    assert len(tags) == 1
-    assert tags[0].get() == file.get(key)
-
-
-def test_anlzfile_get():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    key = "PPTH"
-    tag = file.get_tag(key)
-    assert file.get(key) == tag.get()
-
-
-def test_anlzfile_getall():
-    paths = ANLZ_FILES[0]
-    file = anlz.AnlzFile.parse_file(paths["DAT"])
-    key = "PPTH"
-    tag = file.get_tag(key)
-    values = file.getall(key)
-    assert len(values) == 1
-    assert values[0] == tag.get()

tests/test_config.py

@@ -1,175 +0,0 @@
-# -*- coding: utf-8 -*-
-# Author: Dylan Jones
-# Date:   2023-10-02
-
-import pytest
-import json
-from pathlib import Path
-from pyrekordbox.config import update_config, get_config
-
-RB_SETTING = """<?xml version="1.0" encoding="UTF-8"?>
-<PROPERTIES><VALUE name="masterDbDirectory" val="{db_dir}"/></PROPERTIES>
-"""
-
-RB_OPTIONS = {"options": [["db-path", ""]]}
-
-
-def mock_rekordbox_settings_file(pioneer_app_dir, rb_app_dir):
-    db_dir = pioneer_app_dir / "rekordbox"
-    text = RB_SETTING.format(db_dir=db_dir)
-    file = rb_app_dir / "rekordbox3.settings"
-    file.write_text(text)
-
-
-def mock_rekordbox_options_file(agent_storage_dir, db_path):
-    file = agent_storage_dir / "options.json"
-    options = RB_OPTIONS.copy()
-    options["options"][0][1] = str(db_path)
-    with open(file, "w") as fp:
-        json.dump(options, fp)
-
-
-@pytest.fixture(scope="session")
-def pioneer_app_dir(tmp_path_factory):
-    root = tmp_path_factory.mktemp("Pioneer")
-    rb_dir = root / "rekordbox"
-    rb6_dir = root / "rekordbox6"
-    agent_storage_dir = root / "rekordboxAgent" / "storage"
-    rb_dir.mkdir(parents=True)
-    rb6_dir.mkdir(parents=True)
-    agent_storage_dir.mkdir(parents=True)
-
-    # Mock `rekordbox3.settings` files
-    mock_rekordbox_settings_file(root, rb_dir)
-    mock_rekordbox_settings_file(root, rb6_dir)
-
-    # Mock database files
-    rb5_db = rb_dir / "datafile.edb"
-    rb6_db = rb_dir / "master.db"
-    rb5_db.touch()
-    rb6_db.touch()
-
-    # Mock `options.json` file (RBv6)
-    mock_rekordbox_options_file(agent_storage_dir, rb6_db)
-
-    return root
-
-
-@pytest.fixture(scope="session")
-def pioneer_install_dir(tmp_path_factory):
-    root = tmp_path_factory.mktemp("Pioneer")
-    rb5_dir = root / "rekordbox 5.1.1"
-    rb6_dir = root / "rekordbox 6.1.1"
-    rb5_dir_alt = root / "rekordbox 5.1.2"
-    rb6_dir_alt = root / "rekordbox 6.1.2"
-
-    rb5_dir.mkdir(parents=True)
-    rb6_dir.mkdir(parents=True)
-    rb5_dir_alt.mkdir(parents=True)
-    rb6_dir_alt.mkdir(parents=True)
-    return root
-
-
-def test_pioneer_config(pioneer_install_dir, pioneer_app_dir):
-    update_config(pioneer_install_dir, pioneer_app_dir)
-    install_dir = get_config("pioneer", "install_dir")
-    app_dir = get_config("pioneer", "app_dir")
-
-    assert isinstance(install_dir, Path)
-    assert isinstance(app_dir, Path)
-    assert install_dir == pioneer_install_dir
-    assert app_dir == pioneer_app_dir
-
-
-def test_rb5_config(pioneer_install_dir, pioneer_app_dir):
-    update_config(pioneer_install_dir, pioneer_app_dir)
-    expected_version = "5.1.2"
-
-    app_dir = get_config("rekordbox5", "app_dir")
-    install_dir = get_config("rekordbox5", "install_dir")
-    db_dir = get_config("rekordbox5", "db_dir")
-    db_path = get_config("rekordbox5", "db_path")
-    version = get_config("rekordbox5", "version")
-
-    assert isinstance(install_dir, Path)
-    assert isinstance(app_dir, Path)
-    assert isinstance(db_dir, Path)
-    assert isinstance(db_path, Path)
-    assert install_dir == (pioneer_install_dir / f"rekordbox {expected_version}")
-    assert app_dir == (pioneer_app_dir / "rekordbox")
-    assert db_dir == (pioneer_app_dir / "rekordbox")
-    assert db_path == (pioneer_app_dir / "rekordbox" / "datafile.edb")
-    assert version == expected_version
-
-
-def test_rb6_config(pioneer_install_dir, pioneer_app_dir):
-    update_config(pioneer_install_dir, pioneer_app_dir)
-    expected_version = "6.1.2"
-
-    app_dir = get_config("rekordbox6", "app_dir")
-    install_dir = get_config("rekordbox6", "install_dir")
-    db_dir = get_config("rekordbox6", "db_dir")
-    db_path = get_config("rekordbox6", "db_path")
-    version = get_config("rekordbox6", "version")
-
-    assert isinstance(install_dir, Path)
-    assert isinstance(app_dir, Path)
-    assert isinstance(db_dir, Path)
-    assert isinstance(db_path, Path)
-    assert install_dir == (pioneer_install_dir / f"rekordbox {expected_version}")
-    assert app_dir == (pioneer_app_dir / "rekordbox6")
-    assert db_dir == (pioneer_app_dir / "rekordbox")
-    assert db_path == (pioneer_app_dir / "rekordbox" / "master.db")
-    assert version == expected_version
-
-
-def test_rb5_config_alt(pioneer_install_dir, pioneer_app_dir):
-    # test default: latest version
-    update_config(pioneer_install_dir, pioneer_app_dir)
-    expected_version = "5.1.2"
-
-    app_dir = get_config("rekordbox5", "app_dir")
-    install_dir = get_config("rekordbox5", "install_dir")
-    version = get_config("rekordbox5", "version")
-    assert install_dir == (pioneer_install_dir / f"rekordbox {expected_version}")
-    assert app_dir == (pioneer_app_dir / "rekordbox")
-    assert version == expected_version
-
-    # test alternative version
-    update_config(
-        pioneer_install_dir, pioneer_app_dir, rb5_install_dirname="rekordbox 5.1.1"
-    )
-    expected_version = "5.1.1"
-
-    app_dir = get_config("rekordbox5", "app_dir")
-    install_dir = get_config("rekordbox5", "install_dir")
-    version = get_config("rekordbox5", "version")
-    assert install_dir == (pioneer_install_dir / f"rekordbox {expected_version}")
-    assert app_dir == (pioneer_app_dir / "rekordbox")
-    assert version == expected_version
-
-
-def test_rb6_config_alt(pioneer_install_dir, pioneer_app_dir):
-    # test default: latest version
-    update_config(pioneer_install_dir, pioneer_app_dir)
-    expected_version = "6.1.2"
-
-    app_dir = get_config("rekordbox6", "app_dir")
-    install_dir = get_config("rekordbox6", "install_dir")
-    version = get_config("rekordbox6", "version")
-    assert install_dir == (pioneer_install_dir / f"rekordbox {expected_version}")
-    assert app_dir == (pioneer_app_dir / "rekordbox6")
-    assert version == expected_version
-
-    # test alternative version
-    update_config(
-        pioneer_install_dir, pioneer_app_dir, rb6_install_dirname="rekordbox 6.1.1"
-    )
-    expected_version = "6.1.1"
-
-    app_dir = get_config("rekordbox6", "app_dir")
-    install_dir = get_config("rekordbox6", "install_dir")
-    version = get_config("rekordbox6", "version")
-    assert install_dir == (pioneer_install_dir / f"rekordbox {expected_version}")
-    assert app_dir == (pioneer_app_dir / "rekordbox6")
-    assert version == expected_version