Add test for having a calculate method