bangjunyoung / spiralmatrixsharp Goto Github PK

// It fails with expected output: [| [|3;4|]; [|7;8|]; [|0;0|]; [|0;0|] |]
[<Fact>]
let ``Test ofArray function`` () =
    let output = [| [|1;2|]; [|3;4|]; [|5;6|]; [|7;8|] |] |> array2D
    Assert.Equal(output, Array2D.ofArray 4 2 [|1..8|])

let ofArray (side: int) source =
    if Array.length source <> pown side 2 then
        invalidArg "source" "must be a square array"

    let array2D = Array2D.zeroCreate side side
    source
    |> Array.iteri (fun i elem -> array2D.[i / side, i % side] <- elem)

    array2D

[<Fact>]
let ``Test ofArray function`` () =
    let expected = [| [|1;2;3|]; [|4;5;6|]; [|7;8;9|] |] |> array2D   
    Assert.Equal(expected, Array2D.ofArray 3 [|1..9|])

    Assert.Throws<System.ArgumentException>(
        fun () -> Array2D.ofArray 3 [|1..7|] |> ignore) 
    |> ignore