Author: jcouteau Date: 2009-11-27 15:08:08 +0100 (Fri, 27 Nov 2009) New Revision: 164 Modified: trunk/src/main/java/org/nuiton/j2r/types/RDataFrame.java trunk/src/test/java/org/nuiton/j2r/DataframeTest.java Log: Add a method to get the size of the dataframe and the test associated Modified: trunk/src/main/java/org/nuiton/j2r/types/RDataFrame.java =================================================================== --- trunk/src/main/java/org/nuiton/j2r/types/RDataFrame.java 2009-11-04 10:18:29 UTC (rev 163) +++ trunk/src/main/java/org/nuiton/j2r/types/RDataFrame.java 2009-11-27 14:08:08 UTC (rev 164) @@ -800,4 +800,11 @@ throw new RException("Not supported type"); } } + + public int[] dim() throws RException { + int x = (Integer)engine.eval("dim("+this.variable+")[1]"); + int y = (Integer)engine.eval("dim("+this.variable+")[2]"); + + return (new int[]{x,y}); + } } \ No newline at end of file Modified: trunk/src/test/java/org/nuiton/j2r/DataframeTest.java =================================================================== --- trunk/src/test/java/org/nuiton/j2r/DataframeTest.java 2009-11-04 10:18:29 UTC (rev 163) +++ trunk/src/test/java/org/nuiton/j2r/DataframeTest.java 2009-11-27 14:08:08 UTC (rev 164) @@ -1979,4 +1979,42 @@ engine.setAutoCommit(true); } + + public void testDim (){ + List<String> names = new ArrayList<String>(); + names.add("column1"); + names.add("column2"); + + List<String> rowNames = new ArrayList<String>(); + rowNames.add("row 1"); + rowNames.add("row 2"); + rowNames.add("row 3"); + + List<Double> column1 = new ArrayList<Double>(); + column1.add(3.0); + column1.add(4.5); + column1.add(0.01); + + List<Double> column2 = new ArrayList<Double>(); + column2.add(1.0); + column2.add(5555555555555555555555.0); + column2.add(3.0); + + List<List<? extends Object>> data = new ArrayList<List<? extends Object>>(); + data.add(column1); + data.add(column2); + + RDataFrame testDataFrame = new RDataFrame(engine); + try { + testDataFrame = new RDataFrame(engine, names, rowNames, data, + "test"); + + Assert.assertEquals(3,testDataFrame.dim()[1]); + Assert.assertEquals(2,testDataFrame.dim()[2]); + } catch (RException eee) { + Assert.fail(); + } + + + } }