Add more cost model test cases