enhance unit tests

Author: Tattoo

tests/utest/test_recognize_images.py

@@ -120,8 +120,7 @@ def test_locate_with_valid_reference_folder(self):
                          (u'./reference_images', u'my picture.png'),
                          ('../../tests/utest/reference_images', 'MY PICTURE')):
 
-            ref = path_join(CURDIR, ref)
-            self.lib.reference_folder = ref
+            self.lib.set_reference_folder(path_join(CURDIR, ref))
             self._verify_path_works(img, 'my_picture.png')
 
         self.lib.reference_folder = path_join(CURDIR, 'symbolic_link')

tests/utest/test_screenshot.py

@@ -26,7 +26,7 @@ def tearDown(self):
 
     def _take_screenshot_many_times(self, expected_filename):
         folder = path_join(CURDIR, 'reference_folder')
-        self.lib.screenshot_folder = folder
+        self.lib.set_screenshot_folder(folder)
         for i in range(1, 15):
             self.lib.take_a_screenshot()
             self.mock.screenshot.assert_called_once_with(