def test_get_variables(self): assignment = Assignment() self.assertEqual(0, len(assignment.get_variables())) assignment.set_assignment(self.var1, 1) assignment.set_assignment(self.var2, 2) self.assertSameElements([self.var1, self.var2], assignment.get_variables())