Add basic test for array