Mercurial > hg > auditok
diff tests/test_core.py @ 411:0e938065a2db
AudioRegion as a dataclass
author | Amine Sehili <amine.sehili@gmail.com> |
---|---|
date | Thu, 20 Jun 2024 21:45:08 +0200 |
parents | f56b4d8adfb8 |
children | 9f83c1ecb03b |
line wrap: on
line diff
--- a/tests/test_core.py Wed Jun 19 23:30:18 2024 +0200 +++ b/tests/test_core.py Thu Jun 20 21:45:08 2024 +0200 @@ -1,5 +1,6 @@ import math import os +from pathlib import Path from random import random from tempfile import TemporaryDirectory from unittest.mock import Mock, patch @@ -1362,8 +1363,7 @@ expected_duration_s, expected_duration_ms, ): - meta = {"start": start, "end": expected_end} - region = AudioRegion(data, sampling_rate, sample_width, channels, meta) + region = AudioRegion(data, sampling_rate, sample_width, channels, start) assert region.sampling_rate == sampling_rate assert region.sr == sampling_rate assert region.sample_width == sample_width @@ -1518,9 +1518,7 @@ ) def test_save(format, start, expected): with TemporaryDirectory() as tmpdir: - region = AudioRegion(b"0" * 160, 160, 1, 1) - meta = {"start": start, "end": start + region.duration} - region.meta = meta + region = AudioRegion(b"0" * 160, 160, 1, 1, start) format = os.path.join(tmpdir, format) filename = region.save(format)[len(tmpdir) + 1 :] assert filename == expected @@ -1534,6 +1532,9 @@ with pytest.raises(FileExistsError): region.save(filename, exists_ok=False) + with pytest.raises(FileExistsError): + region.save(Path(filename), exists_ok=False) + @pytest.mark.parametrize( "region, slice_, expected_data",