Add 1d, 2d, 2d with axis tests for std.