def testWriteBlock(self): writer = util.Writer() mod_block = block.ModuleBlock(imputil_test.MockPath(), '__main__', '<test>', '', stmt.FutureFeatures()) writer.write_block(mod_block, 'BODY') output = writer.getvalue() dispatch = 'switch πF.State() {\n\tcase 0:\n\tdefault: panic' self.assertIn(dispatch, output) self.assertIn('return nil, nil\n}', output)
def _MakeModuleBlock(): return block.ModuleBlock('__main__', 'grumpy', 'grumpy/lib', '<test>', [], stmt.FutureFeatures())
def _MakeModuleBlock(): return block.ModuleBlock(imputil_test.MockPath(), '__main__', '<test>', '', stmt.FutureFeatures())
def _MakeModuleBlock(): return block.ModuleBlock('__main__', '<test>', '', stmt.FutureFeatures())