def testTwoClassesDiscrete(self):
# test for #47759
shader = QgsColorRampShader(0, 50, None, QgsColorRampShader.Discrete)
item1 = QgsColorRampShader.ColorRampItem(50, QColor(0, 0, 0))
item2 = QgsColorRampShader.ColorRampItem(float("inf"),
QColor(255, 255, 255))
shader.setColorRampItemList([item1, item2])
color1 = shader.shade(50)
self.assertEqual(color1[1:4], (0, 0, 0))
color2 = shader.shade(50.00000000001)
self.assertEqual(color2[1:4], (255, 255, 255))
def testNan(self):
shader = QgsColorRampShader()
item1 = QgsColorRampShader.ColorRampItem(1, QColor(0, 0, 0))
item2 = QgsColorRampShader.ColorRampItem(2, QColor(255, 255, 255))
shader.setColorRampItemList([item1, item2])
self.assertFalse(shader.shade(float('NaN'))[0])
self.assertFalse(shader.shade(float("inf"))[0])
def testNan(self):
shader = QgsColorRampShader()
item1 = QgsColorRampShader.ColorRampItem(1, QColor(0, 0, 0))
item2 = QgsColorRampShader.ColorRampItem(2, QColor(255, 255, 255))
shader.setColorRampItemList([item1, item2])
self.assertFalse(shader.shade(float('NaN'))[0])
self.assertFalse(shader.shade(float("inf"))[0])
