def test_get_solutions(self):
        """
        Ensure Writer.get_solutions returns a list pertaining to the provided problem number
        """
        testWriter = Writer()
        testWriter._solutions = { 1 : ['sol1', 'sol2', 'sol3'],
                                  2 : ['sol4', 'sol5', 'sol6'],
                                  3 : ['sol7', 'sol8', 'sol9'] 
                                }

        # Ensure correct list is returned
        self.assertEquals(testWriter.get_solutions(2), ['sol4', 'sol5', 'sol6'])
        
        # Ensure invalid number returns an empty list
        self.assertEquals(testWriter.get_solutions(5), [])
    def test_get_all_solutions(self):
        """
        Ensure Writer.get_all_solutions returns a list of all solutions
        """
        testWriter = Writer()

        # Ensure with no solutions an empty list is returned
        self.assertEquals(testWriter.get_all_solutions(), [])

        testWriter._solutions = { 1 : ['sol1', 'sol2', 'sol3'],
                                  2 : ['sol4', 'sol5', 'sol6'],
                                  3 : ['sol7', 'sol8', 'sol9'] 
                                }

        # Ensure all solutions are provided
        self.assertEquals(testWriter.get_all_solutions(), ['sol1', 'sol2', 'sol3',
            'sol4', 'sol5', 'sol6', 'sol7', 'sol8', 'sol9'])