1 files changed, 48 insertions, 0 deletions
diff --git a/Tests/WmapParserTests/StagesParserTests.swift b/Tests/WmapParserTests/StagesParserTests.swift
new file mode 100644
index 0000000..50bab4b
--- /dev/null
+++ b/Tests/WmapParserTests/StagesParserTests.swift
@@ -0,0 +1,48 @@
+import Testing
+
+@testable import WmapParser
+
+@Test func shouldParseStageI() {
+  let source = "[I] 0.25"
+  let result = parse(source)
+  #expect(result.stages.count == 1)
+  #expect(
+    result.stages.first == Stage(stage: .stageI, value: 0.25)
+  )
+}
+
+@Test func shouldParseStageII() {
+  let source = "[ii] 0.5"
+  let result = parse(source)
+  #expect(result.stages.count == 1)
+  #expect(
+    result.stages.first == Stage(stage: .stageII, value: 0.5)
+  )
+}
+
+@Test func shouldParseStageIII() {
+  let source = "[III] 0.75"
+  let result = parse(source)
+  #expect(result.stages.count == 1)
+  #expect(
+    result.stages.first == Stage(stage: .stageIII, value: 0.75)
+  )
+}
+
+@Test func shouldParseStageIV() {
+  let source = "[Iv] 1.0"
+  let result = parse(source)
+  #expect(result.stages.count == 1)
+  #expect(
+    result.stages.first == Stage(stage: .stageIV, value: 1.0)
+  )
+}
+
+@Test func shouldHandleDecimalValues() {
+  let source = "[i] .333"
+  let result = parse(source)
+  #expect(result.stages.count == 1)
+  #expect(
+    result.stages.first == Stage(stage: .stageI, value: 0.333)
+  )
+}