Began to implement test for weight.py
Added test for weight.py