def testOneBandRaster(self):
path = os.path.join(unitTestDataPath('raster'),
'band1_float32_noct_epsg4326.tif')
info = QFileInfo(path)
base_name = info.baseName()
layer = QgsRasterLayer(path, base_name)
self.assertTrue(layer)
combo = QgsRasterBandComboBox()
combo.setLayer(layer)
self.assertEqual(combo.layer(), layer)
self.assertEqual(combo.currentBand(), 1)
self.assertEqual(combo.count(), 1)
combo.setShowNotSetOption(True)
self.assertEqual(combo.currentBand(), 1)
self.assertEqual(combo.count(), 2)
combo.setBand(-1)
self.assertEqual(combo.currentBand(), -1)
combo.setBand(1)
self.assertEqual(combo.currentBand(), 1)
combo.setShowNotSetOption(False)
self.assertEqual(combo.currentBand(), 1)
self.assertEqual(combo.count(), 1)
def testMultiBandRaster(self):
path = os.path.join(unitTestDataPath('raster'),
'band3_float32_noct_epsg4326.tif')
info = QFileInfo(path)
base_name = info.baseName()
layer = QgsRasterLayer(path, base_name)
self.assertTrue(layer)
combo = QgsRasterBandComboBox()
combo.setLayer(layer)
self.assertEqual(combo.layer(), layer)
self.assertEqual(combo.currentBand(), 1)
self.assertEqual(combo.count(), 3)
combo.setBand(2)
self.assertEqual(combo.currentBand(), 2)
combo.setShowNotSetOption(True)
self.assertEqual(combo.currentBand(), 2)
self.assertEqual(combo.count(), 4)
combo.setShowNotSetOption(False)
self.assertEqual(combo.currentBand(), 2)
self.assertEqual(combo.count(), 3)
def testInvalidLayer(self):
layer = QgsRasterLayer('blah', 'blah')
self.assertTrue(layer)
self.assertFalse(layer.isValid())
combo = QgsRasterBandComboBox()
combo.setLayer(layer)
self.assertEqual(combo.count(), 0)
self.assertTrue(combo.isEditable())
signal_spy = QSignalSpy(combo.bandChanged)
combo.setBand(11111)
self.assertEqual(len(signal_spy), 1)
self.assertEqual(signal_spy[-1][0], 11111)
self.assertEqual(combo.currentBand(), 11111)
combo.setBand(-11111)
self.assertEqual(combo.currentBand(), -1)
self.assertEqual(len(signal_spy), 2)
self.assertEqual(signal_spy[-1][0], -1)
# replace with valid layer
path = os.path.join(unitTestDataPath('raster'),
'band3_float32_noct_epsg4326.tif')
info = QFileInfo(path)
base_name = info.baseName()
layer2 = QgsRasterLayer(path, base_name)
self.assertTrue(layer2.isValid())
combo.setBand(2)
self.assertEqual(combo.currentBand(), 2)
self.assertEqual(len(signal_spy), 3)
self.assertEqual(signal_spy[-1][0], 2)
combo.setLayer(layer2)
self.assertEqual(combo.count(), 3)
self.assertFalse(combo.isEditable())
self.assertEqual(combo.currentBand(), 2)
self.assertEqual(len(signal_spy), 3)
# back to invalid
combo.setLayer(layer)
self.assertEqual(combo.count(), 0)
self.assertTrue(combo.isEditable())
self.assertEqual(combo.currentBand(), 2)
self.assertEqual(len(signal_spy), 3)
# with not set option
combo.setShowNotSetOption(True)
combo.setBand(-1)
self.assertEqual(combo.currentBand(), -1)
self.assertEqual(len(signal_spy), 4)
self.assertEqual(signal_spy[-1][0], -1)
combo.setBand(3)
self.assertEqual(combo.currentBand(), 3)
self.assertEqual(len(signal_spy), 5)
self.assertEqual(signal_spy[-1][0], 3)
combo.setBand(-1)
self.assertEqual(combo.currentBand(), -1)
self.assertEqual(len(signal_spy), 6)
self.assertEqual(signal_spy[-1][0], -1)
combo.setLayer(layer2)
self.assertEqual(combo.currentBand(), -1)
self.assertEqual(len(signal_spy), 6)
combo.setLayer(layer)
combo.setCurrentText('bad')
self.assertEqual(combo.currentBand(), -1)
self.assertEqual(len(signal_spy), 6)
self.assertEqual(signal_spy[-1][0], -1)
combo.setCurrentText('5')
self.assertEqual(combo.currentBand(), 5)
self.assertEqual(len(signal_spy), 7)
self.assertEqual(signal_spy[-1][0], 5)
combo.setCurrentText('6.5')
self.assertEqual(combo.currentBand(), -1)
self.assertEqual(len(signal_spy), 8)
self.assertEqual(signal_spy[-1][0], -1)
combo.setCurrentText('5')
self.assertEqual(combo.currentBand(), 5)
self.assertEqual(len(signal_spy), 9)
self.assertEqual(signal_spy[-1][0], 5)
combo.setCurrentText('Not set')
self.assertEqual(combo.currentBand(), -1)
self.assertEqual(len(signal_spy), 10)
self.assertEqual(signal_spy[-1][0], -1)