Improve LeetCode 922 unit tests

joshuamegnauth54 · Mar 15, 2024 · afd9a10 · afd9a10
1 parent 728fe78
commit afd9a10
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/Zig/src/neet/922_array_parity_2.zig b/Zig/src/neet/922_array_parity_2.zig
@@ -73,14 +73,22 @@ test "LeetCode example 2" {
     try expectEqualSlices(u8, &expected, actual);
 }
 
-test "unbalanced parity fails" {
+test "unbalanced parity (even) fails" {
     const input = [_]u8{ 4, 6, 8, 10 };
     const expected = ArrayParityError.NotEnoughOdds;
 
-    const actual = sort_array_parity_2(u8, std.testing.allocator, &input) catch |err| {
-        try expectError(expected, err);
-    };
+    const actual = sort_array_parity_2(u8, std.testing.allocator, &input);
+    errdefer std.testing.allocator.free(actual catch unreachable);
+
+    try expectError(expected, actual);
+}
+
+test "unbalanced parity (odd) fails" {
+    const input = [_]u8{ 3, 5, 7, 9 };
+    const expected = ArrayParityError.NotEnoughEvens;
+
+    const actual = sort_array_parity_2(u8, std.testing.allocator, &input);
+    errdefer std.testing.allocator.free(actual catch unreachable);
 
-    std.testing.allocator.free(actual);
-    @panic("Expected error `NotEnoughOdds`");
+    try expectError(expected, actual);
 }