54 行
1.8 KiB
Python
54 行
1.8 KiB
Python
from yandex_music import RotorSettings
|
|
|
|
|
|
class TestRotorSettings:
|
|
language = 'not-russian'
|
|
diversity = 'default'
|
|
mood = 2
|
|
energy = 2
|
|
mood_energy = 'fun'
|
|
|
|
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_none(self, client):
|
|
assert RotorSettings.de_json({}, client) is None
|
|
|
|
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):
|
|
a = RotorSettings(self.language, self.diversity)
|
|
b = RotorSettings('', self.diversity)
|
|
c = RotorSettings(self.language, self.diversity)
|
|
|
|
assert a != b
|
|
assert hash(a) != hash(b)
|
|
assert a is not b
|
|
|
|
assert a == c
|