57 lines
1.4 KiB
Zig
57 lines
1.4 KiB
Zig
const std = @import("std");
|
|
const root = @import("root.zig");
|
|
const compileNzsl = root.compileNzsl;
|
|
const case = root.case;
|
|
|
|
test "Simple while loop" {
|
|
const allocator = std.testing.allocator;
|
|
const base = @mod(case.random(f32), 5.0);
|
|
const iterations = 5;
|
|
|
|
var expected = base;
|
|
for (1..iterations) |i| {
|
|
expected *= @floatFromInt(i);
|
|
}
|
|
|
|
const shader = try std.fmt.allocPrint(
|
|
allocator,
|
|
\\ [nzsl_version("1.1")]
|
|
\\ [feature(float64)]
|
|
\\ module;
|
|
\\
|
|
\\ struct FragOut
|
|
\\ {{
|
|
\\ [location(0)] color: vec4[f32]
|
|
\\ }}
|
|
\\
|
|
\\ [entry(frag)]
|
|
\\ fn main() -> FragOut
|
|
\\ {{
|
|
\\ let value = f32({d});
|
|
\\ let i = 1;
|
|
\\ while (i < {d})
|
|
\\ {{
|
|
\\ value *= f32(i);
|
|
\\ i += 1;
|
|
\\ }}
|
|
\\ let output: FragOut;
|
|
\\ output.color = vec4[f32](value, value, value, value);
|
|
\\ return output;
|
|
\\ }}
|
|
,
|
|
.{
|
|
base,
|
|
iterations,
|
|
},
|
|
);
|
|
defer allocator.free(shader);
|
|
const code = try compileNzsl(allocator, shader);
|
|
defer allocator.free(code);
|
|
try case.expect(.{
|
|
.source = code,
|
|
.expected_outputs = &.{
|
|
std.mem.asBytes(&[_]f32{ expected, expected, expected, expected }),
|
|
},
|
|
});
|
|
}
|