Fix tests to just call assertEqual rather than unnecessarily type-specific methods e.g. assertSetEqual