add unittest for density optimization