tests: Add argument to allow specifying which directories to test