Add method generator tests