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.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])): self.assertEqual(v, p) self.assertSetEqual(set(self.view[two]), set([3])) self.assertSetEqual(set(self.param_index[two]), set([0, 5])) self.view.add(two, np.array([0])) self.assertSetEqual(set(self.view[two]), set([0, 3])) self.assertSetEqual(set(self.param_index[two]), set([0, 2, 5])) self.view.clear() for v, p in zip(self.view.properties_for(np.r_[:6]), self.param_index.properties_for(np.r_[2:2 + 6])): self.assertEqual(v, p) self.assertEqual(v, []) param_index = ParameterIndexOperations() param_index.add(one, [3, 9]) param_index.add(two, [0, 5]) param_index.add(three, [2, 4, 7, 10]) view2 = ParameterIndexOperationsView(param_index, 2, 8) self.view.update(view2) for [i, v], [i2, v2] in zip(sorted(param_index.items()), sorted(self.param_index.items())): self.assertEqual(i, i2) np.testing.assert_equal(v, v2)
def test_parameter_index_operations(self): pio = ParameterIndexOperations( dict(test1=np.array([4, 3, 1, 6, 4]), test2=np.r_[2:130])) piov = ParameterIndexOperationsView(pio, 20, 250) #py3 fix #self.assertListDictEquals(dict(piov.items()), dict(piov.copy().iteritems())) self.assertListDictEquals(dict(piov.items()), dict(piov.copy().items())) #py3 fix #self.assertListDictEquals(dict(pio.iteritems()), dict(pio.copy().items())) self.assertListDictEquals(dict(pio.items()), dict(pio.copy().items())) self.assertArrayListEquals(pio.copy().indices(), pio.indices()) self.assertArrayListEquals(piov.copy().indices(), piov.indices()) with tempfile.TemporaryFile('w+b') as f: pickle.dump(pio, f) f.seek(0) pio2 = pickle.load(f) self.assertListDictEquals(pio._properties, pio2._properties) f = tempfile.TemporaryFile('w+b') with f: pickle.dump(piov, f) f.seek(0) pio2 = pickle.load(f) #py3 fix #self.assertListDictEquals(dict(piov.items()), dict(pio2.iteritems())) self.assertListDictEquals(dict(piov.items()), dict(pio2.items()))
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 setUp(self): self.param_index = ParameterIndexOperations() self.param_index.add(one, [3, 9]) self.param_index.add(two, [0, 5]) self.param_index.add(three, [2, 4, 7, 10]) self.view = ParameterIndexOperationsView(self.param_index, 2, 6)