def test_staging(self): network = Constructor() network.set_grid(5, 5, 100) network.delete_connection(0, 1) with self.assertRaises( AssertionError): # no moving after deleting street network.move_horizontal_line(0, 10) with self.assertRaises(AssertionError): network.move_vertical_line(0, 10) network.modify_adjacency(10, 0.5, 0.5) with self.assertRaises(AssertionError): # no moving after modifying network.move_vertical_line(0, 10) with self.assertRaises(AssertionError): network.move_horizontal_line(0, 10) with self.assertRaises(AssertionError): # no deleting after modifying network.delete_connection(0, 1)
def test_move_vertical_line(self): # Test normal usage network1 = Constructor() network1.set_grid(4, 4, 100) network1.move_vertical_line(3, 30) positions1 = network1.get_positions() for i in range(3, 16, 4): self.assertEqual(positions1[i][0], 330) network2 = Constructor() network2.set_grid(2, 2, 100) network2.move_vertical_line(1, 100) positions2 = network2.get_positions() for i in range(1, 4, 2): self.assertEqual(positions2[i][0], 200) # Test wrong usage with self.assertRaises(ValueError): network1.move_vertical_line(-1, 0) # no such vertical line with self.assertRaises(ValueError): network2.move_vertical_line(2, 0) # no such vertical line