apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-10 12:32:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

[IndexOps] added new tests for index operations

Browse source
This commit is contained in:
Max Zwiessele 2014-10-31 17:10:39 +00:00
parent 0080d4b044
commit 4183b6665e
1 changed files with 55 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

81
GPy/testing/index_operations_tests.py
View file

@ -14,9 +14,9 @@ class Test(unittest.TestCase):
def setUp(self): def setUp(self):
self.param_index = ParameterIndexOperations() self.param_index = ParameterIndexOperations()
self.param_index.add(one, [3]) self.param_index.add(one, [3,9])
self.param_index.add(two, [0,5]) self.param_index.add(two, [0,5])
self.param_index.add(three, [2,4,7]) self.param_index.add(three, [2,4,7,10])
self.view = ParameterIndexOperationsView(self.param_index, 2, 6) self.view = ParameterIndexOperationsView(self.param_index, 2, 6)
def test_clear(self): def test_clear(self):
@ -24,39 +24,63 @@ class Test(unittest.TestCase):
self.assertDictEqual(self.param_index._properties, {}) self.assertDictEqual(self.param_index._properties, {})
def test_remove(self): def test_remove(self):
removed = self.param_index.remove(three, np.r_[3:10]) removed = self.param_index.remove(three, np.r_[3:13])
self.assertListEqual(removed.tolist(), [4, 7]) self.assertListEqual(removed.tolist(), [4,7,10])
self.assertListEqual(self.param_index[three].tolist(), [2]) self.assertListEqual(self.param_index[three].tolist(), [2])
removed = self.param_index.remove(one, [1]) removed = self.param_index.remove(one, [1])
self.assertListEqual(removed.tolist(), []) self.assertListEqual(removed.tolist(), [])
self.assertListEqual(self.param_index[one].tolist(), [3]) self.assertListEqual(self.param_index[one].tolist(), [3,9])
self.assertListEqual(self.param_index.remove('not in there', []).tolist(), []) self.assertListEqual(self.param_index.remove('not in there', []).tolist(), [])
removed = self.param_index.remove(one, [9]) removed = self.param_index.remove(one, [9])
self.assertListEqual(removed.tolist(), [9])
self.assertListEqual(self.param_index[one].tolist(), [3]) self.assertListEqual(self.param_index[one].tolist(), [3])
self.assertListEqual(self.param_index.remove('not in there', [2,3,4]).tolist(), []) self.assertListEqual(self.param_index.remove('not in there', [2,3,4]).tolist(), [])
self.assertListEqual(self.view.remove('not in there', [2,3,4]).tolist(), []) self.assertListEqual(self.view.remove('not in there', [2,3,4]).tolist(), [])
def test_shift_left(self): def test_shift_left(self):
self.view.shift_left(0, 2) self.view.shift_left(0, 2)
self.assertListEqual(self.param_index[three].tolist(), [2,5]) self.assertListEqual(self.param_index[three].tolist(), [2,5,8])
self.assertListEqual(self.param_index[two].tolist(), [0,3]) self.assertListEqual(self.param_index[two].tolist(), [0,3])
self.assertListEqual(self.param_index[one].tolist(), []) self.assertListEqual(self.param_index[one].tolist(), [7])
def test_shift_right(self):
self.view.shift_right(3, 2)
self.assertListEqual(self.param_index[three].tolist(), [2,4,9])
self.assertListEqual(self.param_index[two].tolist(), [0,7])
self.assertListEqual(self.param_index[one].tolist(), [3])
def test_index_view(self):
#======================================================================= #=======================================================================
# 0 1 2 3 4 5 6 7 8 9 # 0 1 2 3 4 5 6 7 8 9 10
# one # one
# two two # two two
# three three three # three three three
# view: [0 1 2 3 4 5 ] # view: [0 1 2 3 4 5 ]
#======================================================================= #=======================================================================
self.assertListEqual(self.view[three].tolist(), [0,3])
self.assertListEqual(self.view[two].tolist(), [1])
self.assertListEqual(self.view[one].tolist(), [5])
self.param_index.shift_left(7, 1)
#=======================================================================
# 0 1 2 3 4 5 6 7 8 9 10
#
# two two
# three three three
# view: [0 1 2 3 4 5 ]
#=======================================================================
self.assertListEqual(self.param_index[three].tolist(), [2,5,7])
self.assertListEqual(self.param_index[two].tolist(), [0,3])
self.assertListEqual(self.param_index[one].tolist(), [])
self.assertListEqual(self.view[three].tolist(), [0,3,5])
self.assertListEqual(self.view[two].tolist(), [1])
self.assertListEqual(self.view[one].tolist(), [])
def test_shift_right(self):
self.view.shift_right(3, 2)
self.assertListEqual(self.param_index[three].tolist(), [2,4,9,12])
self.assertListEqual(self.param_index[two].tolist(), [0,7])
self.assertListEqual(self.param_index[one].tolist(), [3,11])
def test_index_view(self):
#=======================================================================
# 0 1 2 3 4 5 6 7 8 9 10
# one one
# two two
# three three three three
# view: [0 1 2 3 4 5 ]
#=======================================================================
self.view = ParameterIndexOperationsView(self.param_index, 2, 6) self.view = ParameterIndexOperationsView(self.param_index, 2, 6)
self.assertSetEqual(set(self.view.properties()), set([one, two, three])) self.assertSetEqual(set(self.view.properties()), set([one, two, three]))
for v,p in zip(self.view.properties_for(np.r_[:6]), self.param_index.properties_for(np.r_[2:2+6])): for v,p in zip(self.view.properties_for(np.r_[:6]), self.param_index.properties_for(np.r_[2:2+6])):
@ -71,14 +95,26 @@ class Test(unittest.TestCase):
self.assertEqual(v, p) self.assertEqual(v, p)
self.assertEqual(v, []) self.assertEqual(v, [])
param_index = ParameterIndexOperations() param_index = ParameterIndexOperations()
param_index.add(one, [3]) param_index.add(one, [3,9])
param_index.add(two, [0,5]) param_index.add(two, [0,5])
param_index.add(three, [2,4,7]) param_index.add(three, [2,4,7,10])
view2 = ParameterIndexOperationsView(param_index, 2, 6) view2 = ParameterIndexOperationsView(param_index, 2, 8)
self.view.update(view2) self.view.update(view2)
for [i,v],[i2,v2] in zip(sorted(param_index.items()), sorted(self.param_index.items())): for [i,v],[i2,v2] in zip(sorted(param_index.items()), sorted(self.param_index.items())):
self.assertEqual(i, i2) self.assertEqual(i, i2)
self.assertTrue(np.all(v == v2)) np.testing.assert_equal(v, v2)
def test_view_of_view(self):
#=======================================================================
# 0 1 2 3 4 5 6 7 8 9 10
# one one
# two two
# three three three three
# view: [0 1 2 3 4 5 ]
# view2: [0 1 2 3 4 5 ]
#=======================================================================
view2 = ParameterIndexOperationsView(self.view, 2, 6)
view2.shift_right(0, 2)
def test_indexview_remove(self): def test_indexview_remove(self):
removed = self.view.remove(two, [3]) removed = self.view.remove(two, [3])
@ -86,11 +122,10 @@ class Test(unittest.TestCase):
removed = self.view.remove(three, np.r_[:5]) removed = self.view.remove(three, np.r_[:5])
self.assertListEqual(removed.tolist(), [0, 2]) self.assertListEqual(removed.tolist(), [0, 2])
def test_misc(self): def test_misc(self):
for k,v in self.param_index.copy()._properties.iteritems(): for k,v in self.param_index.copy()._properties.iteritems():
self.assertListEqual(self.param_index[k].tolist(), v.tolist()) self.assertListEqual(self.param_index[k].tolist(), v.tolist())
self.assertEqual(self.param_index.size, 6) self.assertEqual(self.param_index.size, 8)
self.assertEqual(self.view.size, 5) self.assertEqual(self.view.size, 5)
def test_print(self): def test_print(self):