46 行
1.6 KiB
Python
46 行
1.6 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from yandex_music import RotorSettings
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope='class')
|
||
|
|
def rotor_settings():
|
||
|
|
return RotorSettings(TestRotorSettings.language, TestRotorSettings.diversity, TestRotorSettings.mood,
|
||
|
|
TestRotorSettings.energy, TestRotorSettings.mood_energy)
|
||
|
|
|
||
|
|
|
||
|
|
class TestRotorSettings:
|
||
|
|
language = None
|
||
|
|
diversity = None
|
||
|
|
mood = None
|
||
|
|
energy = None
|
||
|
|
mood_energy = None
|
||
|
|
|
||
|
|
def test_expected_values(self, rotor_settings):
|
||
|
|
assert rotor_settings.language == self.language
|
||
|
|
assert rotor_settings.diversity == self.diversity
|
||
|
|
assert rotor_settings.mood == self.mood
|
||
|
|
assert rotor_settings.energy == self.energy
|
||
|
|
assert rotor_settings.mood_energy == self.mood_energy
|
||
|
|
|
||
|
|
def test_de_json_required(self, client):
|
||
|
|
json_dict = {'language': self.language, 'diversity': self.diversity}
|
||
|
|
rotor_settings = RotorSettings.de_json(json_dict, client)
|
||
|
|
|
||
|
|
assert rotor_settings.language == self.language
|
||
|
|
assert rotor_settings.diversity == self.diversity
|
||
|
|
|
||
|
|
def test_de_json_all(self, client):
|
||
|
|
json_dict = {'language': self.language, 'diversity': self.diversity, 'mood': self.mood, 'energy': self.energy,
|
||
|
|
'mood_energy': self.mood_energy}
|
||
|
|
rotor_settings = RotorSettings.de_json(json_dict, client)
|
||
|
|
|
||
|
|
assert rotor_settings.language == self.language
|
||
|
|
assert rotor_settings.diversity == self.diversity
|
||
|
|
assert rotor_settings.mood == self.mood
|
||
|
|
assert rotor_settings.energy == self.energy
|
||
|
|
assert rotor_settings.mood_energy == self.mood_energy
|
||
|
|
|
||
|
|
def test_equality(self):
|
||
|
|
pass
|