ErrorNoInternet/ErrorNoCord
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

test: check for short formatting cases

Browse Source
This commit is contained in:
Ryan 2025-01-08 09:25:48 -05:00
parent 6afbce5d8f
commit 07b3bde22d
Signed by: ErrorNoInternet
GPG Key ID: 2486BFB7B1E6A4A3
1 changed files with 88 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

130
tests/test_format_duration.py
View File

@ -6,56 +6,102 @@ import youtubedl
class TestFormatDuration(unittest.TestCase): class TestFormatDuration(unittest.TestCase):
def test_youtubedl(self): def test_youtubedl(self):
self.assertEqual(youtubedl.format_duration(0), "00:00") def f(s):
self.assertEqual(youtubedl.format_duration(0.5), "00:00") return youtubedl.format_duration(s)
self.assertEqual(youtubedl.format_duration(60.5), "01:00")
self.assertEqual(youtubedl.format_duration(1), "00:01") self.assertEqual(f(0), "00:00")
self.assertEqual(youtubedl.format_duration(60), "01:00") self.assertEqual(f(0.5), "00:00")
self.assertEqual(youtubedl.format_duration(60 + 30), "01:30") self.assertEqual(f(60.5), "01:00")
self.assertEqual(youtubedl.format_duration(60 * 60), "01:00:00") self.assertEqual(f(1), "00:01")
self.assertEqual(youtubedl.format_duration(60 * 60 + 30), "01:00:30") self.assertEqual(f(60), "01:00")
self.assertEqual(f(60 + 30), "01:30")
self.assertEqual(f(60 * 60), "01:00:00")
self.assertEqual(f(60 * 60 + 30), "01:00:30")
def test_utils(self): def test_utils(self):
self.assertEqual(utils.format_duration(0), "") def f(s):
self.assertEqual(utils.format_duration(60 * 60 * 24 * 7), "1 week") return utils.format_duration(s)
self.assertEqual(utils.format_duration(60 * 60 * 24 * 21), "3 weeks")
self.assertEqual(f(0), "")
self.assertEqual(f(60 * 60 * 24 * 7), "1 week")
self.assertEqual(f(60 * 60 * 24 * 21), "3 weeks")
self.assertEqual( self.assertEqual(
utils.format_duration((60 * 60 * 24 * 21) - 1), f((60 * 60 * 24 * 21) - 1),
"2 weeks, 6 days, 23 hours, 59 minutes, 59 seconds", "2 weeks, 6 days, 23 hours, 59 minutes, 59 seconds",
) )
self.assertEqual(utils.format_duration(60), "1 minute") self.assertEqual(f(60), "1 minute")
self.assertEqual(utils.format_duration(60 * 2), "2 minutes") self.assertEqual(f(60 * 2), "2 minutes")
self.assertEqual(utils.format_duration(60 * 59), "59 minutes") self.assertEqual(f(60 * 59), "59 minutes")
self.assertEqual(utils.format_duration(60 * 60), "1 hour") self.assertEqual(f(60 * 60), "1 hour")
self.assertEqual(utils.format_duration(60 * 60 * 2), "2 hours") self.assertEqual(f(60 * 60 * 2), "2 hours")
self.assertEqual(utils.format_duration(1), "1 second") self.assertEqual(f(1), "1 second")
self.assertEqual(utils.format_duration(60 + 5), "1 minute, 5 seconds") self.assertEqual(f(60 + 5), "1 minute, 5 seconds")
self.assertEqual(utils.format_duration(60 * 60 + 30), "1 hour, 30 seconds") self.assertEqual(f(60 * 60 + 30), "1 hour, 30 seconds")
self.assertEqual( self.assertEqual(f(60 * 60 + 60 + 30), "1 hour, 1 minute, 30 seconds")
utils.format_duration(60 * 60 + 60 + 30), "1 hour, 1 minute, 30 seconds" self.assertEqual(f(60 * 60 * 24 * 7 + 30), "1 week, 30 seconds")
)
self.assertEqual(
utils.format_duration(60 * 60 * 24 * 7 + 30), "1 week, 30 seconds"
)
def test_utils_natural(self): def test_utils_natural(self):
def format(seconds: int): def f(s):
return utils.format_duration(seconds, natural=True) return utils.format_duration(s, natural=True)
self.assertEqual(format(0), "") self.assertEqual(f(0), "")
self.assertEqual(format(60 * 60 * 24 * 7), "1 week") self.assertEqual(f(60 * 60 * 24 * 7), "1 week")
self.assertEqual(format(60 * 60 * 24 * 21), "3 weeks") self.assertEqual(f(60 * 60 * 24 * 21), "3 weeks")
self.assertEqual( self.assertEqual(
format((60 * 60 * 24 * 21) - 1), f((60 * 60 * 24 * 21) - 1),
"2 weeks, 6 days, 23 hours, 59 minutes and 59 seconds", "2 weeks, 6 days, 23 hours, 59 minutes and 59 seconds",
) )
self.assertEqual(format(60), "1 minute") self.assertEqual(f(60), "1 minute")
self.assertEqual(format(60 * 2), "2 minutes") self.assertEqual(f(60 * 2), "2 minutes")
self.assertEqual(format(60 * 59), "59 minutes") self.assertEqual(f(60 * 59), "59 minutes")
self.assertEqual(format(60 * 60), "1 hour") self.assertEqual(f(60 * 60), "1 hour")
self.assertEqual(format(60 * 60 * 2), "2 hours") self.assertEqual(f(60 * 60 * 2), "2 hours")
self.assertEqual(format(1), "1 second") self.assertEqual(f(1), "1 second")
self.assertEqual(format(60 + 5), "1 minute and 5 seconds") self.assertEqual(f(60 + 5), "1 minute and 5 seconds")
self.assertEqual(format(60 * 60 + 30), "1 hour and 30 seconds") self.assertEqual(f(60 * 60 + 30), "1 hour and 30 seconds")
self.assertEqual(format(60 * 60 + 60 + 30), "1 hour, 1 minute and 30 seconds") self.assertEqual(f(60 * 60 + 60 + 30), "1 hour, 1 minute and 30 seconds")
self.assertEqual(format(60 * 60 * 24 * 7 + 30), "1 week and 30 seconds") self.assertEqual(f(60 * 60 * 24 * 7 + 30), "1 week and 30 seconds")
def test_utils_short(self):
def f(s):
return utils.format_duration(s, short=True)
self.assertEqual(f(0), "")
self.assertEqual(f(60 * 60 * 24 * 7), "1w")
self.assertEqual(f(60 * 60 * 24 * 21), "3w")
self.assertEqual(
f((60 * 60 * 24 * 21) - 1),
"2w 6d 23h 59m 59s",
)
self.assertEqual(f(60), "1m")
self.assertEqual(f(60 * 2), "2m")
self.assertEqual(f(60 * 59), "59m")
self.assertEqual(f(60 * 60), "1h")
self.assertEqual(f(60 * 60 * 2), "2h")
self.assertEqual(f(1), "1s")
self.assertEqual(f(60 + 5), "1m 5s")
self.assertEqual(f(60 * 60 + 30), "1h 30s")
self.assertEqual(f(60 * 60 + 60 + 30), "1h 1m 30s")
self.assertEqual(f(60 * 60 * 24 * 7 + 30), "1w 30s")
def test_utils_natural_short(self):
def f(s):
return utils.format_duration(s, natural=True, short=True)
self.assertEqual(f(0), "")
self.assertEqual(f(60 * 60 * 24 * 7), "1w")
self.assertEqual(f(60 * 60 * 24 * 21), "3w")
self.assertEqual(
f((60 * 60 * 24 * 21) - 1),
"2w 6d 23h 59m and 59s",
)
self.assertEqual(f(60), "1m")
self.assertEqual(f(60 * 2), "2m")
self.assertEqual(f(60 * 59), "59m")
self.assertEqual(f(60 * 60), "1h")
self.assertEqual(f(60 * 60 * 2), "2h")
self.assertEqual(f(1), "1s")
self.assertEqual(f(60 + 5), "1m and 5s")
self.assertEqual(f(60 * 60 + 30), "1h and 30s")
self.assertEqual(f(60 * 60 + 60 + 30), "1h 1m and 30s")
self.assertEqual(f(60 * 60 * 24 * 7 + 30), "1w and 30s")