Refactor tests to use TemporalMemoryMonitorMixin