def testCreate(self): package = block.Package('foo/bar/baz') self.assertEqual(package.name, 'foo/bar/baz') self.assertEqual(package.alias, 'π_fooΓbarΓbaz')
def testCreateGrump(self): package = block.Package('foo/bar/baz', 'myalias') self.assertEqual(package.name, 'foo/bar/baz') self.assertEqual(package.alias, 'myalias')
def testWriteImportBlockImportsSorted(self): writer = util.Writer() imports = {name: block.Package(name) for name in ('a', 'b', 'c')} writer.write_import_block(imports) self.assertEqual(writer.getvalue(), 'import (\n\tπ_a "a"\n\tπ_b "b"\n\tπ_c "c"\n)\n')