bindings/test_galearn_pdm.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16


import galearn_pdm
import numpy

def test_one():
    
    inp = numpy.ones(shape=10, dtype=numpy.uint8)
    out = numpy.zeros(shape=10, dtype=numpy.int16)    

    galearn_pdm.process(inp, out)
    print(out)
    assert out[0] == 2
    assert out[-1] == 2

if __name__ == '__main__':
    test_one()