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])