KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > test > prefuse > data > io > CSVTableReaderTest


1 package test.prefuse.data.io;
2
3 import java.io.ByteArrayInputStream JavaDoc;
4 import java.io.InputStream JavaDoc;
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 // public void testReadTableString() {
16
// String datasrc = "/congressional_elections.csv";
17
// CSVTableReader ctr = new CSVTableReader();
18
// try {
19
// URL dataurl = CSVTableReader.class.getResource(datasrc);
20
// Table t = ctr.readTable(dataurl);
21
// assertTrue(t.getRowCount() > 9000);
22
// } catch ( Exception e ) {
23
// fail("Error occurred: " +e);
24
// }
25
// }
26

27     /*
28      * Test method for 'edu.berkeley.guir.prefuse.data.io.CSVTableReader.readTable(InputStream)'
29      */

30     public void testReadTableInputStream() {
31         // prepare data
32
byte[] data = CSV_DATA.getBytes();
33         InputStream JavaDoc is = new ByteArrayInputStream JavaDoc(data);
34         
35         // parse data
36
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         // text-dump
48
if (verbose) System.out.println("-- Data Types -------------");
49         for (int c = 0, idx = -1; c < t.getColumnCount(); ++c) {
50             String JavaDoc 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 JavaDoc 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 // // interface
77
// JFrame f = new JFrame("CSV Loader Test");
78
// f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
79
// JTable jt = new JTable(t) {
80
// TableCellRenderer defr = new DefaultTableCellRenderer();
81
// public TableCellRenderer getCellRenderer(int r, int c) {
82
// return defr;
83
// }
84
// };
85
// JScrollPane jsp = new JScrollPane(jt);
86
// f.getContentPane().add(jsp);
87
// f.pack();
88
// f.setVisible(true);
89
}
90
91 }
92
Popular Tags