Add TEST_DIRS enviroment for constrain test directories in rakefile

What do you think?