def testAddScheme(self): """Test adding a scheme to a registry""" registry = QgsColorSchemeRegistry() self.assertEqual(len(registry.schemes()), 0) recentScheme = QgsRecentColorScheme() registry.addColorScheme(recentScheme) self.assertEqual(len(registry.schemes()), 1)
def testRemoveScheme(self): """Test removing a scheme from a registry""" registry = QgsColorSchemeRegistry() self.assertEqual(len(registry.schemes()), 0) recentScheme = QgsRecentColorScheme() registry.addColorScheme(recentScheme) self.assertEqual(len(registry.schemes()), 1) # remove the scheme registry.removeColorScheme(recentScheme) self.assertEqual(len(registry.schemes()), 0) # try removing a scheme not in the registry self.assertFalse(registry.removeColorScheme(recentScheme))