Add unittest for volume