def test_can_write_allocation_to_file(self, AllocationWriter, os_path):
self.file_path = AllocationWriter.write_allocation(print_file=True)
self.assertTrue(os_path.isfile(self.file_path))
def test_can_write_allocation_to_stdio(self):
AllocationWriter.write_allocation(print_stdio=True)
if not hasattr(sys.stdout, "getvalue"):
self.fail("need to run test in buffered mode")
output = sys.stdout.getvalue().strip()
self.assertIn('Room 1 (OFFICE)', output)