def test_add_associated_nodes(self):
"""test for add_associated_nodes"""
nodes_layer = self.jawa_nodes_layer
add_associated_nodes(nodes_layer, THRESHOLD)
features = nodes_layer.getFeatures()
expected_attributes = [
[0, 5, u'upstream', None, None, 1, 0],
[1, 5, u'downstream', u'2', u'5', 1, 2],
[2, 6, u'upstream', None, u'1,5', 1, 2],
[3, 6, u'downstream', None, None, 0, 1],
[4, 4, u'upstream', u'6', u'9', 2, 1],
[5, 4, u'downstream', u'2', u'1', 1, 2],
[6, 3, u'upstream', u'4', u'9', 2, 1],
[7, 3, u'downstream', None, None, 0, 1],
[8, 2, u'upstream', None, u'11', 1, 1],
[9, 2, u'downstream', u'4,6', None, 2, 1],
[10, 1, u'upstream', None, None, 1, 0],
[11, 1, u'downstream', u'8', None, 1, 1]
]
i = 0
for feature in features:
i += 1
message = feature.attributes()
self.assertIn(feature.attributes(), expected_attributes, message)
message = ('There should be %s features but I got %s.' % (
len(expected_attributes), i))
self.assertEqual(len(expected_attributes), i, message)
def test_check_associated_attributes(self):
"""Test for check_associated_attributes"""
nodes_layer = get_temp_shapefile_layer(JAWA_NODES_SHP, 'nodes')
message = 'Should be False.'
assert not check_associated_attributes(nodes_layer), message
add_associated_nodes(nodes_layer, THRESHOLD)
message = 'Should be True.'
assert check_associated_attributes(nodes_layer), message
def setUpClass(cls):
cls.sungai_layer = get_temp_shapefile_layer(
JAWA_SHP, 'sungai_di_jawa')
cls.dgn_layer = get_temp_shapefile_layer(
DGN_SHP, 'dgn_test ')
cls.jawa_nodes_layer = get_temp_shapefile_layer(
JAWA_NODES_SHP, 'nodes')
cls.dgn_nodes_layer = get_temp_shapefile_layer(
DGN_NODES_SHP, 'nodes')
cls.prepared_nodes_layer = get_temp_shapefile_layer(
JAWA_NODES_SHP, 'prepared_nodes')
add_associated_nodes(cls.prepared_nodes_layer, THRESHOLD)
