def test_paste_vulnerability(self):
mw = MainWindow(os.path.join(EXAMPLES, 'vm.xml'))
# 4 columns for 4 columns
dvs = mw.widget.tv['tableDiscreteVulnerabilitySet']
added = mw.paste_text(dvs, '2\t3\t4\t5')
self.assertEqual([str(dvs.table[i]) for i in added], ['2,3,4,5'])
# 2 columns instead of 4
dvs.tableView.setCurrentIndex(index(0, 0))
added = mw.paste_text(dvs, '2\t3')
self.assertEqual(added, [])
# 2 columns for 2 columns
dv = mw.widget.tv['tableDiscreteVulnerability']
added = mw.paste_text(dv, '2\t3')
self.assertEqual([str(dv.table[i]) for i in added], ['vm1,2,3'])
# 3 columns instead of 2
with self.assertRaises(ValueError):
mw.paste_text(dv, '2\t3\t4') # got 3 columns expected 2
# 3 columns for 3 columns
dv.tableView.setCurrentIndex(index(0, 0))
dvd = mw.widget.tv['tableDiscreteVulnerabilityData']
added = mw.paste_text(dvd, '2\t3\t4')
self.assertEqual([str(dvd.table[i]) for i in added],
['vm1,M1_LR_C,2,3,4'])
def test_current_record_tv0(self):
tv = self.widget.tv[0]
tv.tableView.setCurrentIndex(index(1, 0))
self.assertEqual(
self.widget.tv[0].current_record(),
['vm2', 'Buildings', 'Economic_loss', 'SA(0.34)'])
