1 package test.prefuse.data.io; 2 3 import java.io.ByteArrayInputStream ; 4 import java.io.InputStream ; 5 6 import junit.framework.TestCase; 7 import prefuse.data.Table; 8 import prefuse.data.io.CSVTableReader; 9 import prefuse.data.io.DataIOException; 10 import test.prefuse.TestConfig; 11 import test.prefuse.data.TableTestData; 12 13 public class CSVTableReaderTest extends TestCase implements TableTestData { 14 15 27 30 public void testReadTableInputStream() { 31 byte[] data = CSV_DATA.getBytes(); 33 InputStream is = new ByteArrayInputStream (data); 34 35 CSVTableReader ctr = new CSVTableReader(); 37 Table t = null; 38 try { 39 t = ctr.readTable(is); 40 } catch ( DataIOException e ) { 41 e.printStackTrace(); 42 fail("Data Read Exception"); 43 } 44 45 boolean verbose = TestConfig.verbose(); 46 47 if (verbose) System.out.println("-- Data Types -------------"); 49 for (int c = 0, idx = -1; c < t.getColumnCount(); ++c) { 50 String name = t.getColumnType(c).getName(); 51 if ( (idx=name.lastIndexOf('.')) >= 0 ) 52 name = name.substring(idx+1); 53 assertEquals(t.getColumnType(c), TYPES[c]); 54 if (verbose) System.out.print(name + "\t"); 55 } 56 if (verbose) System.out.println(); 57 58 if (verbose) System.out.println(); 59 60 if (verbose) System.out.println("-- Table Data -------------"); 61 for (int c = 0; c < t.getColumnCount(); ++c) { 62 if (verbose) System.out.print(t.getColumnName(c) + "\t"); 63 assertEquals(t.getColumnName(c), HEADERS[c]); 64 } 65 if (verbose) System.out.println(); 66 for (int r = 0; r < t.getRowCount(); ++r) { 67 for (int c = 0; c < t.getColumnCount(); ++c) { 68 Object o = t.get(r, c); 69 if (verbose) System.out.print(o + "\t"); 70 assertEquals(TABLE[c][r], o); 71 } 72 if (verbose) System.out.println(); 73 } 74 if (verbose) System.out.println(); 75 76 } 90 91 } 92 | Popular Tags |