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_all(self): mw = MainWindow() mw.new_vulnerability_model() mw.new_fragility_model_continuous() mw.new_fragility_model_discrete() mw.new_exposure_model_population() mw.new_exposure_model_buildings()