Adding support for the test_files attribute