Added equality assertion into diagnostics test