mirror of
https://github.com/beetbox/beets.git
synced 2026-01-04 23:12:51 +01:00
fix tests for cached weights loading
This commit is contained in:
parent
b814168fb6
commit
7f9e301428
1 changed files with 16 additions and 0 deletions
|
|
@ -104,6 +104,12 @@ def _make_trackinfo():
|
|||
TrackInfo(u'three', None, u'some artist', length=1, index=3),
|
||||
]
|
||||
|
||||
def _clear_weights():
|
||||
"""Hack around the lazy descriptor used to cache weights for
|
||||
Distance calculations.
|
||||
"""
|
||||
Distance.__dict__['_weights'].computed = False
|
||||
|
||||
class DistanceTest(_common.TestCase):
|
||||
def test_add(self):
|
||||
dist = Distance()
|
||||
|
|
@ -196,6 +202,8 @@ class DistanceTest(_common.TestCase):
|
|||
def test_distance(self):
|
||||
config['match']['distance_weights']['album'] = 2.0
|
||||
config['match']['distance_weights']['medium'] = 1.0
|
||||
_clear_weights()
|
||||
|
||||
dist = Distance()
|
||||
dist.add('album', 0.5)
|
||||
dist.add('media', 0.25)
|
||||
|
|
@ -209,6 +217,8 @@ class DistanceTest(_common.TestCase):
|
|||
def test_max_distance(self):
|
||||
config['match']['distance_weights']['album'] = 3.0
|
||||
config['match']['distance_weights']['medium'] = 1.0
|
||||
_clear_weights()
|
||||
|
||||
dist = Distance()
|
||||
dist.add('album', 0.5)
|
||||
dist.add('medium', 0.0)
|
||||
|
|
@ -219,6 +229,8 @@ class DistanceTest(_common.TestCase):
|
|||
config['match']['distance_weights']['source'] = 1.0
|
||||
config['match']['distance_weights']['album'] = 2.0
|
||||
config['match']['distance_weights']['medium'] = 1.0
|
||||
_clear_weights()
|
||||
|
||||
dist = Distance()
|
||||
dist.add('source', 0.0)
|
||||
dist.add('album', 0.5)
|
||||
|
|
@ -236,6 +248,8 @@ class DistanceTest(_common.TestCase):
|
|||
def test_raw_distance(self):
|
||||
config['match']['distance_weights']['album'] = 3.0
|
||||
config['match']['distance_weights']['medium'] = 1.0
|
||||
_clear_weights()
|
||||
|
||||
dist = Distance()
|
||||
dist.add('album', 0.5)
|
||||
dist.add('medium', 0.25)
|
||||
|
|
@ -245,6 +259,8 @@ class DistanceTest(_common.TestCase):
|
|||
def test_items(self):
|
||||
config['match']['distance_weights']['album'] = 4.0
|
||||
config['match']['distance_weights']['medium'] = 2.0
|
||||
_clear_weights()
|
||||
|
||||
dist = Distance()
|
||||
dist.add('album', 0.1875)
|
||||
dist.add('medium', 0.75)
|
||||
|
|
|
|||
Loading…
Reference in a new issue