can you add test cases for how you expect this to be used?