public void testPyImportHandling() throws Exception { Document doc = new Document("from xxx import yyy"); PyImportsHandling importsHandling = new PyImportsHandling(doc); Iterator<ImportHandle> it = importsHandling.iterator(); assertTrue(it.hasNext()); ImportHandle next = it.next(); assertEquals("from xxx import yyy", next.importFound); assertEquals(0, next.startFoundLine); assertEquals(0, next.endFoundLine); assertFalse(it.hasNext()); }
public void testPyImportHandling5() throws Exception { Document doc = new Document("import threading;\n"); PyImportsHandling importsHandling = new PyImportsHandling(doc, false); Iterator<ImportHandle> it = importsHandling.iterator(); assertTrue(it.hasNext()); ImportHandle next = it.next(); assertEquals("import threading;", next.importFound); assertEquals(0, next.startFoundLine); assertEquals(0, next.endFoundLine); assertTrue(!it.hasNext()); }
public void testPyImportHandling3() throws Exception { Document doc = new Document("from ...a.b import b\nfrom xxx.bbb \\\n import yyy\n"); PyImportsHandling importsHandling = new PyImportsHandling(doc); Iterator<ImportHandle> it = importsHandling.iterator(); assertTrue(it.hasNext()); ImportHandle next = it.next(); assertEquals("from ...a.b import b", next.importFound); assertEquals(0, next.startFoundLine); assertEquals(0, next.endFoundLine); next = it.next(); assertEquals("from xxx.bbb \\\n import yyy", next.importFound); assertEquals(1, next.startFoundLine); assertEquals(2, next.endFoundLine); assertTrue(!it.hasNext()); }