Added tests for unit weights