def check_workflow(self, dag, name):
expected_rootdir = \
"{}/sharemind_expected".format(os.path.dirname(os.path.realpath(__file__)))
sm_cfg = SharemindCodeGenConfig()
cfg = CodeGenConfig('cfg').with_sharemind_config(sm_cfg)
cg = SharemindCodeGen(cfg, dag, 1)
actual = cg._generate('code', '/tmp')[1]['miner']
with open(expected_rootdir + '/{}'.format(name), 'r') as f:
expected = f.read()
self.assertEqual(expected, actual)
def check_workflow(self, dag, name, use_leaky_ops=True):
self.maxDiff = None
expected_rootdir = \
"{}/sharemind_expected".format(os.path.dirname(os.path.realpath(__file__)))
sm_cfg = SharemindCodeGenConfig()
cfg = CodeGenConfig('cfg').with_sharemind_config(sm_cfg)
cfg.use_leaky_ops = use_leaky_ops
cg = SharemindCodeGen(cfg, dag, 1)
actual = cg._generate('code', '/tmp')[1]['miner']
with open(expected_rootdir + '/{}'.format(name),
'r') as f_specific, open(
expected_rootdir + '/{}'.format("base"), 'r') as f_base:
expected_base = f_base.read()
expected_specific = f_specific.read()
expected = expected_base + expected_specific
self.assertEqual(expected, actual)