我有一个简单的单元测试,在这里我检查是否可以用稍微不同的参数实例化我的Tensorflow类。对于@pytest.mark.parametrize来说,这似乎是一个很好的用例。例如,当我在以下代码上运行pytest时:
class TestBasicRewardNet(tf.test.TestCase):为了解决这个问题,我试着摆脱类包装器,但这让我错过了一些自动的Tensorflow测试初始化。特别是,现在每个BasicRewardNet都构建在同一个TensorFlow图中,所以我需要添加一个变量范围