Add basic test setup