`timescale 1ns / 1ps

module Test_PriorityEncoder();
    logic[7:0] data_in;
    logic[2:0] data_out;
    PriorityEncoder#(.N(3)) encoder(data_in, data_out);
    initial begin
        data_in = 'b00000001;
        for (int i = 0; i < 8; i++) begin
            assert (data_out == i + 1) else $error("[One-hot] Failed at " + i);
            #1
            data_in = data_in << 'd1;
        end
        #1
        data_in = 'b00101111;
        assert (data_out == 'd5) else $error("[Manual entry] Failed at " + 5);
        #1
        data_in = 'b10101010;
        assert (data_out == 'd7) else $error("[Manual entry] Failed at " + 7);
    end
endmodule