Skip to content

Commit

Permalink
tidy up object ref unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
jakeret committed Feb 24, 2016
1 parent 093d5e8 commit 21e9395
Showing 1 changed file with 11 additions and 9 deletions.
20 changes: 11 additions & 9 deletions test/test_object.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,25 +83,26 @@ def fkt(self):
hfkt = hope.jit(fkt)

t1 = Test()
arr1 = t1.fkt()
arr2 = t1.hfkt()
assert np.all(arr1 == arr2)
arr = t1.fkt()
harr = t1.hfkt()
assert np.all(arr == harr)
assert t1.member is harr

def test_member_reference_scalar():
class Test(object):
def __init__(self, n=10):
self.member = n

def fkt(self):
arr = self.member
return arr
a = self.member
return a

hfkt = hope.jit(fkt)

t1 = Test()
arr1 = t1.fkt()
arr2 = t1.hfkt()
assert np.all(arr1 == arr2)
a = t1.fkt()
ha = t1.hfkt()
assert a == ha

def test_member_reference_view():
class Test(object):
Expand All @@ -116,6 +117,7 @@ def fkt(self):

t1 = Test()
t1.fkt()
assert t1.member is not t1.out
assert np.all(t1.member == t1.out)

def test_member_reference_member():
Expand All @@ -131,7 +133,7 @@ def fkt(self):

t1 = Test()
t1.fkt()
assert np.all(t1.member == t1.out)
assert t1.member is t1.out


if __name__ == '__main__':
Expand Down

0 comments on commit 21e9395

Please sign in to comment.