Add assertEqual and assertThrows