Unit tests are exactly for code that is often rewritten, because it ensures that whatever interface still behaves the same, regardless of the implementation. This a large portion of the point of unit tests: not for testing the initial implementation but confirming that any subsequent implementation behaves the same.