Store test weight in the wrapper
To allow the test runner to access the weight of an individual test, we store the weight as an attribute of the wrapper.
Though the interaction between the two should be simplified down the line, this is probably the cleanest way to do it with the current design.
Needed to fix this issue.