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')
