def testQgsColorBrewerColorRampV2(self):
# test color brewer color ramps
r = QgsColorBrewerColorRamp('OrRd', 6)
self.assertEqual(r.type(), 'colorbrewer')
self.assertEqual(r.schemeName(), 'OrRd')
self.assertEqual(r.count(), 6)
self.assertEqual(r.value(0), 0)
self.assertEqual(r.value(1), 0.2)
self.assertEqual(r.value(2), 0.4)
self.assertEqual(r.value(3), 0.6)
self.assertEqual(r.value(4), 0.8)
self.assertEqual(r.value(5), 1)
self.assertTrue(not r.color(-1).isValid())
self.assertTrue(not r.color(6).isValid())
self.assertEqual(r.color(0), QColor(254, 240, 217))
self.assertEqual(r.color(0.2), QColor(253, 212, 158))
self.assertEqual(r.color(0.4), QColor(253, 187, 132))
self.assertEqual(r.color(0.6), QColor(252, 141, 89))
self.assertEqual(r.color(0.8), QColor(227, 74, 51))
self.assertEqual(r.color(1.0), QColor(179, 0, 0))
# try using an invalid scheme name
bad = QgsColorBrewerColorRamp('badscheme', 6)
self.assertFalse(bad.color(0).isValid())
self.assertEqual(bad.value(1), 0)
# test creating from properties
props = r.properties()
fromProps = QgsColorBrewerColorRamp.create(props)
self.assertEqual(fromProps.type(), 'colorbrewer')
self.assertEqual(fromProps.schemeName(), 'OrRd')
self.assertEqual(fromProps.count(), 6)
self.assertEqual(fromProps.color(0), QColor(254, 240, 217))
self.assertEqual(fromProps.color(0.2), QColor(253, 212, 158))
self.assertEqual(fromProps.color(0.4), QColor(253, 187, 132))
self.assertEqual(fromProps.color(0.6), QColor(252, 141, 89))
self.assertEqual(fromProps.color(0.8), QColor(227, 74, 51))
self.assertEqual(fromProps.color(1.0), QColor(179, 0, 0))
# test cloning ramp
cloned = r.clone()
self.assertEqual(cloned.type(), 'colorbrewer')
self.assertEqual(cloned.schemeName(), 'OrRd')
self.assertEqual(cloned.count(), 6)
self.assertEqual(cloned.color(0), QColor(254, 240, 217))
self.assertEqual(cloned.color(0.2), QColor(253, 212, 158))
self.assertEqual(cloned.color(0.4), QColor(253, 187, 132))
self.assertEqual(cloned.color(0.6), QColor(252, 141, 89))
self.assertEqual(cloned.color(0.8), QColor(227, 74, 51))
self.assertEqual(cloned.color(1.0), QColor(179, 0, 0))
# set scheme name
r.setSchemeName('Reds')
self.assertEqual(r.schemeName(), 'Reds')
self.assertEqual(r.count(), 6)
self.assertEqual(r.color(0), QColor(254, 229, 217))
self.assertEqual(r.color(0.2), QColor(252, 187, 161))
self.assertEqual(r.color(0.4), QColor(252, 146, 114))
self.assertEqual(r.color(0.6), QColor(251, 106, 74))
self.assertEqual(r.color(0.8), QColor(222, 45, 38))
self.assertEqual(r.color(1.0), QColor(165, 15, 21))
# test invert function
r.invert()
self.assertEqual(r.color(0), QColor(165, 15, 21))
self.assertEqual(r.color(0.2), QColor(222, 45, 38))
self.assertEqual(r.color(1), QColor(254, 229, 217))
r.invert()
# set colors
r.setColors(3)
self.assertEqual(r.colors(), 3)
self.assertEqual(r.count(), 3)
self.assertEqual(r.color(0), QColor(254, 224, 210))
self.assertEqual(r.color(0.5), QColor(252, 146, 114))
self.assertEqual(r.color(1.0), QColor(222, 45, 38))
# test static members
names = QgsColorBrewerColorRamp.listSchemeNames()
self.assertTrue('Reds' in names and 'OrRd' in names)
self.assertEqual(
len(QgsColorBrewerColorRamp.listSchemeVariants('bad scheme')), 0)
variants = QgsColorBrewerColorRamp.listSchemeVariants('Reds')
self.assertEqual(variants, [3, 4, 5, 6, 7, 8, 9])
def testQgsColorBrewerColorRampV2(self):
# test color brewer color ramps
r = QgsColorBrewerColorRamp('OrRd', 6)
self.assertEqual(r.type(), 'colorbrewer')
self.assertEqual(r.schemeName(), 'OrRd')
self.assertEqual(r.count(), 6)
self.assertEqual(r.value(0), 0)
self.assertEqual(r.value(1), 0.2)
self.assertEqual(r.value(2), 0.4)
self.assertEqual(r.value(3), 0.6)
self.assertEqual(r.value(4), 0.8)
self.assertEqual(r.value(5), 1)
self.assertTrue(not r.color(-1).isValid())
self.assertTrue(not r.color(6).isValid())
self.assertEqual(r.color(0), QColor(254, 240, 217))
self.assertEqual(r.color(0.2), QColor(253, 212, 158))
self.assertEqual(r.color(0.4), QColor(253, 187, 132))
self.assertEqual(r.color(0.6), QColor(252, 141, 89))
self.assertEqual(r.color(0.8), QColor(227, 74, 51))
self.assertEqual(r.color(1.0), QColor(179, 0, 0))
# try using an invalid scheme name
bad = QgsColorBrewerColorRamp('badscheme', 6)
self.assertFalse(bad.color(0).isValid())
self.assertEqual(bad.value(1), 0)
# test creating from properties
props = r.properties()
fromProps = QgsColorBrewerColorRamp.create(props)
self.assertEqual(fromProps.type(), 'colorbrewer')
self.assertEqual(fromProps.schemeName(), 'OrRd')
self.assertEqual(fromProps.count(), 6)
self.assertEqual(fromProps.color(0), QColor(254, 240, 217))
self.assertEqual(fromProps.color(0.2), QColor(253, 212, 158))
self.assertEqual(fromProps.color(0.4), QColor(253, 187, 132))
self.assertEqual(fromProps.color(0.6), QColor(252, 141, 89))
self.assertEqual(fromProps.color(0.8), QColor(227, 74, 51))
self.assertEqual(fromProps.color(1.0), QColor(179, 0, 0))
# test cloning ramp
cloned = r.clone()
self.assertEqual(cloned.type(), 'colorbrewer')
self.assertEqual(cloned.schemeName(), 'OrRd')
self.assertEqual(cloned.count(), 6)
self.assertEqual(cloned.color(0), QColor(254, 240, 217))
self.assertEqual(cloned.color(0.2), QColor(253, 212, 158))
self.assertEqual(cloned.color(0.4), QColor(253, 187, 132))
self.assertEqual(cloned.color(0.6), QColor(252, 141, 89))
self.assertEqual(cloned.color(0.8), QColor(227, 74, 51))
self.assertEqual(cloned.color(1.0), QColor(179, 0, 0))
# set scheme name
r.setSchemeName('Reds')
self.assertEqual(r.schemeName(), 'Reds')
self.assertEqual(r.count(), 6)
self.assertEqual(r.color(0), QColor(254, 229, 217))
self.assertEqual(r.color(0.2), QColor(252, 187, 161))
self.assertEqual(r.color(0.4), QColor(252, 146, 114))
self.assertEqual(r.color(0.6), QColor(251, 106, 74))
self.assertEqual(r.color(0.8), QColor(222, 45, 38))
self.assertEqual(r.color(1.0), QColor(165, 15, 21))
# test invert function
r.invert()
self.assertEqual(r.color(0), QColor(165, 15, 21))
self.assertEqual(r.color(0.2), QColor(222, 45, 38))
self.assertEqual(r.color(1), QColor(254, 229, 217))
r.invert()
# set colors
r.setColors(3)
self.assertEqual(r.colors(), 3)
self.assertEqual(r.count(), 3)
self.assertEqual(r.color(0), QColor(254, 224, 210))
self.assertEqual(r.color(0.5), QColor(252, 146, 114))
self.assertEqual(r.color(1.0), QColor(222, 45, 38))
# test static members
names = QgsColorBrewerColorRamp.listSchemeNames()
self.assertTrue('Reds' in names and 'OrRd' in names)
self.assertEqual(len(QgsColorBrewerColorRamp.listSchemeVariants('bad scheme')), 0)
variants = QgsColorBrewerColorRamp.listSchemeVariants('Reds')
self.assertEqual(variants, [3, 4, 5, 6, 7, 8, 9])
def getColorBrewColorRampGnYlRd():
colorRamp = QgsColorBrewerColorRamp(schemeName='RdYlGn', inverted=True)
return colorRamp
