1 files changed, 32 insertions, 0 deletions
diff --git a/config/mpv/scripts/subs2srs/test.lua b/config/mpv/scripts/subs2srs/test.lua
new file mode 100644
index 0000000..d1291d2
--- /dev/null
+++ b/config/mpv/scripts/subs2srs/test.lua
@@ -0,0 +1,32 @@
+local helpers = require('helpers')
+
+local function assert_equals(expected, actual)
+    if expected ~= actual then
+        error(string.format("TEST FAILED: Expected '%s', got '%s'", expected, actual))
+    end
+end
+
+local function test_get_episode_number()
+    local test_cases = {
+      { nil, "A Whisker Away.mkv" },
+      { nil, "[Placeholder] Gekijouban SHIROBAKO [Ma10p_1080p][x265_flac]" },
+      { "06", "[Placeholder] Sono Bisque Doll wa Koi wo Suru - 06 [54E495D0]" },
+      { "02", "(Hi10)_Kobayashi-san_Chi_no_Maid_Dragon_-_02_(BD_1080p)_(Placeholder)_(12C5D2B4)" },
+      { "01", "[Placeholder] Koi to Yobu ni wa Kimochi Warui - 01 (1080p) [D517C9F0]" },
+      { "01", "[Placeholder] Tsukimonogatari 01 [BD 1080p x264 10-bit FLAC] [5CD88145]" },
+      { "01", "[Placeholder] 86 - Eighty Six - 01 (1080p) [1B13598F]" },
+      { "00", "[Placeholder] Fate Stay Night - Unlimited Blade Works - 00 (BD 1080p Hi10 FLAC) [95590B7F]" },
+      { "01", "House, M.D. S01E01 Pilot - Everybody Lies (1080p x265 Placeholder)" },
+      { "165", "A Generic Episode-165" }
+    }
+
+    for _, case in pairs(test_cases) do
+      local _, _, episode_num = helpers.get_episode_number(case[2])
+      assert_equals(case[1], episode_num)
+    end
+end
+
+-- Runs tests
+test_get_episode_number()
+
+os.exit(print("Tests passed"))