修补Python单元测试中的模块级功能（生成器）

Question

我有一个模块级方法（它是一个生成器）。我想提供该方法的自定义实现（不仅仅是在测试另一个类时返回值。我还需要在测试时更改类的一些参数（实例变量）。

模块x：

def zone_file_iterator():
    pass 

模块y：

from x import zone_file_iterator()

class PleaseTestMe(CachedSingleTon):
      def __init__():
        self.a

      @classmethod
      def load_data(cls):
        """
        do some thing with large data file.
        """
        zone_file_iterator()

为了测试上面的类，我相信在Python Unittest中使用Mock Framework，我们可以做这样的事情吗？如果这是错误的，请纠正我。

   @patch('y.PleaseTestMe.a', customvalue)
   @patch('x.zone_file_iterator', x.custome_small_data_iterator)
   def test_skip_grams():
      """
      Instantiate the class and write Assert Logic here.
      """
      pass