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