add test for partial trace