1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
|
package org.python.core;
import junit.framework.TestCase;
import org.python.util.PythonInterpreter;
public class WrappedIntegerTest extends TestCase {
// Simulate the use case where you want to expose some (possibly mutable)
// java int field to an interpreter without having to set the value to a new
// PyInteger each time it changes.
@SuppressWarnings("serial")
static class WrappedInteger extends PyInteger {
public WrappedInteger() {
super(0);
}
private int mutableValue;
@Override
public int getValue() {
return mutableValue;
}
public void setMutableValue(final int newValue) {
mutableValue = newValue;
}
}
private PythonInterpreter interp;
private WrappedInteger a, b;
@Override
protected void setUp() throws Exception {
interp = new PythonInterpreter(new PyStringMap(), new PySystemState());
a = new WrappedInteger();
b = new WrappedInteger();
a.setMutableValue(13);
b.setMutableValue(17);
interp.set("a", a);
interp.set("b", b);
}
public void testAdd() {
interp.exec("c = a + b");
assertEquals(new PyInteger(30), interp.get("c"));
b.setMutableValue(18);
interp.exec("c = a + b");
assertEquals(new PyInteger(31), interp.get("c"));
}
public void testDiv() {
interp.exec("c = a / float(b)");
assertEquals(new PyFloat(13 / 17.), interp.get("c"));
}
public void testMod() {
interp.exec("c = b % a");
assertEquals(new PyInteger(4), interp.get("c"));
}
}
|