improve add_unit_test()