def fromFile(self, initializeWithRandomVels=True):
# Parse from file
self.nodes, self.links = parse_input(self.file)
# Set the position of each node
square = math.ceil(math.sqrt(len(self.nodes)))
for i in range(square):
for j in range(square):
if (i * square + j < len(self.nodes)):
self.nodes[i * square +
j].pos = (int((WIDTH) / (square) * i + 20),
int((HEIGHT) / (square) * j + 20))
if (initializeWithRandomVels):
self.nodes[i * square +
j].vel = generateRandomVelocity()
else:
self.nodes[i * square + j].vel = (0, 0)
self.nodes[i * square + j].color = createColorFromString(
self.nodes[square * i + j].text)
def test_exit(self):
self.assertEqual(parse.parse_input("exit"), (False, None))
def test_history(self):
self.assertEqual(parse.parse_input("history"), (True, None))
def test_operations(self):
self.assertEqual(parse.parse_input("operations"), (True, None))
def test_help(self):
self.assertEqual(parse.parse_input("help"), (True, None))
def test_inputparsing(self):
self.assertEqual(parse.parse_input("5 exp 2 /"),
(True, str(math.exp(5) / 2)))
