HW2 chore: moved test cases to a separate file