testing/conftest.py at main · beatsbyte/testing · GitHub

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
62
63
64
65
66
67
68
69
70
71
72
# conftest.py
import pytest
from pydub import AudioSegment
import numpy as np
import os
import aiohttp
import asyncio
from aiohttp.payload import BytesPayload
import requests
@pytest.fixture
async def send_music():
    async def _send_music(url, file_path, id):
        try:
            with open(file_path, "rb") as file:
                file_content = file.read()
        except FileNotFoundError:
            pytest.fail(f"Test MP3 file not found: {file_path}")


        with aiohttp.MultipartWriter() as data:
            payload = BytesPayload(
                file_content,
                content_type="audio/mpeg",
                filename="test_input.mp3"
            )

            part = data.append_payload(payload)
            part.set_content_disposition('form-data', name='file', filename="test_input.mp3")

        headers = {
            "Content-Type": f"multipart/form-data; boundary={data.boundary}",
        }


        async with aiohttp.ClientSession() as session:
                async with session.post(f'{url}/v1/compress', data=data, headers=headers) as response:
                    if response.status != 200:
                        pytest.fail(f"Request failed with status: {response.status}")
                    ogg_content = await response.read()

        # ogg_content = await response.read()
        output_file_path = f"data/test_output_{id}.ogg"
        with open(output_file_path, "wb") as output_file:
            output_file.write(ogg_content)

        return output_file_path

    return _send_music


@pytest.fixture
def audio2npArray():
    def _audio2npArray(audio_path):
        audio = AudioSegment.from_file(audio_path)
        audio_common_type = "data/audio.wav"
        audio.export(audio_common_type, format="wav")
        audio = AudioSegment.from_wav(audio_common_type).set_channels(1)
        return np.array(audio.get_array_of_samples())
    return _audio2npArray

@pytest.fixture
def compare_audio():
    def _compare_audio(np_array1, np_array2):
        min_len = min(len(np_array1), len(np_array2))
        np_array1 = np_array1[:min_len]
        np_array2 = np_array2[:min_len]

        correlation = np.corrcoef(np_array1, np_array2)[0, 1]
        correlation = max(correlation, 0)
        return correlation * 100

    return _compare_audio