mirror of
https://github.com/beetbox/beets.git
synced 2025-12-10 02:22:25 +01:00
a6e5201ff3
This is 'the big one', which touches every file so that it all conforms to the given standard.
72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
"""Tests for the 'the' plugin"""
|
|
|
|
|
|
import unittest
|
|
from test import _common
|
|
|
|
from beets import config
|
|
from beetsplug.the import FORMAT, PATTERN_A, PATTERN_THE, ThePlugin
|
|
|
|
|
|
class ThePluginTest(_common.TestCase):
|
|
def test_unthe_with_default_patterns(self):
|
|
self.assertEqual(ThePlugin().unthe("", PATTERN_THE), "")
|
|
self.assertEqual(
|
|
ThePlugin().unthe("The Something", PATTERN_THE), "Something, The"
|
|
)
|
|
self.assertEqual(ThePlugin().unthe("The The", PATTERN_THE), "The, The")
|
|
self.assertEqual(
|
|
ThePlugin().unthe("The The", PATTERN_THE), "The, The"
|
|
)
|
|
self.assertEqual(
|
|
ThePlugin().unthe("The The X", PATTERN_THE), "The X, The"
|
|
)
|
|
self.assertEqual(ThePlugin().unthe("the The", PATTERN_THE), "The, the")
|
|
self.assertEqual(
|
|
ThePlugin().unthe("Protected The", PATTERN_THE), "Protected The"
|
|
)
|
|
self.assertEqual(ThePlugin().unthe("A Boy", PATTERN_A), "Boy, A")
|
|
self.assertEqual(ThePlugin().unthe("a girl", PATTERN_A), "girl, a")
|
|
self.assertEqual(ThePlugin().unthe("An Apple", PATTERN_A), "Apple, An")
|
|
self.assertEqual(
|
|
ThePlugin().unthe("An A Thing", PATTERN_A), "A Thing, An"
|
|
)
|
|
self.assertEqual(
|
|
ThePlugin().unthe("the An Arse", PATTERN_A), "the An Arse"
|
|
)
|
|
self.assertEqual(
|
|
ThePlugin().unthe("TET - Travailleur", PATTERN_THE),
|
|
"TET - Travailleur",
|
|
)
|
|
|
|
def test_unthe_with_strip(self):
|
|
config["the"]["strip"] = True
|
|
self.assertEqual(
|
|
ThePlugin().unthe("The Something", PATTERN_THE), "Something"
|
|
)
|
|
self.assertEqual(ThePlugin().unthe("An A", PATTERN_A), "A")
|
|
|
|
def test_template_function_with_defaults(self):
|
|
ThePlugin().patterns = [PATTERN_THE, PATTERN_A]
|
|
self.assertEqual(ThePlugin().the_template_func("The The"), "The, The")
|
|
self.assertEqual(ThePlugin().the_template_func("An A"), "A, An")
|
|
|
|
def test_custom_pattern(self):
|
|
config["the"]["patterns"] = ["^test\\s"]
|
|
config["the"]["format"] = FORMAT
|
|
self.assertEqual(
|
|
ThePlugin().the_template_func("test passed"), "passed, test"
|
|
)
|
|
|
|
def test_custom_format(self):
|
|
config["the"]["patterns"] = [PATTERN_THE, PATTERN_A]
|
|
config["the"]["format"] = "{1} ({0})"
|
|
self.assertEqual(ThePlugin().the_template_func("The A"), "The (A)")
|
|
|
|
|
|
def suite():
|
|
return unittest.TestLoader().loadTestsFromName(__name__)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main(defaultTest="suite")
|