pyjnius/tests/test_lambdas.py

27 lines
852 B
Python
Raw Normal View History

import unittest
from jnius import autoclass, JavaException
class TestLambdas(unittest.TestCase):
def testJavaUtilFunction(self):
Arrays = autoclass("java.util.Arrays")
Collectors = autoclass("java.util.stream.Collectors")
#J: List<Integer> numbers = Arrays.asList(1, 2, 3);
numbers = autoclass('java.util.ArrayList')()
numbers.add(1)
numbers.add(2)
numbers.add(3)
#J: List<Integer> squares = numbers.stream().map( i -> i*i).collect(Collectors.toList());
squares = numbers.stream().map(lambda i : i * i).collect(Collectors.toList())
self.assertEqual(len(squares), len(numbers))
self.assertEqual(squares[0], 1)
self.assertEqual(squares[1], 4)
self.assertEqual(squares[2], 9)
if __name__ == '__main__':
unittest.main()