Add unit test for unweighted case

Still need to check values