python - 测试 python 中某个字段是否已初始化

Question

我正在尝试用 Python 编写一个测试，检查我正在编写的类中的方法是否设置某些 Hdf 文件中数据集的属性值。逻辑如下:通过传递 h5py.File 实例来构造该类的实例，然后一种方法在此文件内创建一个数据集。在下一步中，我有另一种方法来设置该数据集的某些属性。

我试图测试的是我的类方法create_attributes(self,attributes)是否将字段hdf_file[dset_name].attrs[attr_name]设置为某个值传入变量attributes。但是，我想避免实际创建 Hdf 文件。到目前为止，我已经尝试模拟 hdf 文件的实例并使用它。最小的工作代码示例如下:

import h5py

class TestSomething:
    @mock.patch('h5py.File')
    def test_if_attr_is_initialized(self,mock_hdf):
        # Here I would like to call a function that basically executes
        # the following line:
        mock_hdf['test_dset'].attrs['test_field']='value'

        # Then I want to check if the attribute field has been assigned
        assert mock_hdf['test_dset'].attrs['test_field']=='value'

有人可以帮我找到正确的方法来检查 hdf 文件中的属性是否设置正确吗？任何帮助将不胜感激，我对所有模拟技术都是新手。

编辑:

在下面，我为该类以及二战所要求的相应测试提供了一个最小的代码示例:

import h5py

class HdfWriter():
     def __init__(self,hdf_file):
         self.hdf_file=hdf_file

     def create_attrs(self,attributes):
         dset_name=attributes.keys()[0]
         attrs=attributes[dset_name]

         for key in attrs:
             self.hdf_file[dset_name].attrs[key]=attrs[key]

请注意，使用真正的 hdf 文件，我首先必须创建一个数据集，但我想将其留到另一个测试。以下测试应该只是检查，对于具有数据集 test_dset 的假设 hdf 文件，是否写入了该数据集的属性:

import h5py
import HdfWriter

class TestSomething:
    @mock.patch('h5py.File')
    def test_if_attr_is_initialized(self,mock_hdf):
        writer=hw.HdfWriter(mock_hdf)
        attr={'test_dset':{'test_field':'test_value'}}
        writer.create_attrs(attr)

        assert writer.hdf_file['test_dset'].attrs['test_field']=='value'

Answer 1

模拟h5py.File

class HdfWriter():
     def __init__(self,hdf_file):
         self.hdf_file=hdf_file

     def create_attrs(self,attributes):
         dset_name=attributes.keys()[0]
         attrs=attributes[dset_name]

         for key in attrs:
             self.hdf_file[dset_name].attrs[key]=attrs[key]

就 create_attrs 方法而言，hdf_file 表现为字典，返回的对象也表现为字典。该文档非常清楚地解释了如何模拟字典。

您需要一个具有类似于字典的 attrs 属性的模拟:

import mock

attrs_d = {}
def setattrs(name, value):
##    print 'setattrs', name, value
    attrs_d[name] = value
def getattrs(name):
##    print 'getattrs', name
    return attrs_d[name]
mock = mock.MagicMock()
mock.attrs.__setitem__.side_effect = setattrs
mock.attrs.__getitem__.side_effect = getattrs

您需要一个 hdf_file 的模拟，它的行为类似于字典，并将返回上面创建的模拟对象。

hdf_d = {'test_dset':mock}
def getitem(name):
##    print 'getitem', name
    return hdf_d[name]
def setitem(name, value):
    hdf_d[name] = value
mock_hdf = mock.MagicMock()
mock_hdf.__getitem__.side_effect = getitem
mock_hdf.__setitem__.side_effect = setitem

hdf_d，按照实现方式，仅适用于键'test_dset'。根据您的需求，无论名称参数如何，getitems 只返回 mock 可能会更好。

def test_if_attr_is_initialized(mock_hdf):
    writer=HdfWriter(mock_hdf)
    attr={'test_dset':{'test_field':'test_value'}}
    writer.create_attrs(attr)
    print writer.hdf_file['test_dset'].attrs['test_field'], '==', attr['test_dset']['test_field']

    assert writer.hdf_file['test_dset'].attrs['test_field']=='test_value'

test_if_attr_is_initialized(mock_hdf)

>>> 
test_value == test_value
>>> 

这应该足以测试create_attrs，但它可能不是最佳的 - 也许有人会加入一些改进。