Example #1
0
 def test_lock(self):
     st1 = Struct({_k: [_i] for _i, _k in enumerate(['a', 'b', 'c'])})
     st1._lock()
     with self.assertRaises(AttributeError):
         st1.a = 1
     with self.assertRaises(AttributeError):
         st1.d = 1
     with self.assertRaises(AttributeError):
         st1['a'] = 1
     with self.assertRaises(AttributeError):
         st1['d'] = 1
     with self.assertRaises(AttributeError):
         del st1.b
     with self.assertRaises(AttributeError):
         st1.col_remove('c')
     with self.assertRaises(AttributeError):
         _ = st1.col_pop('c')
     with self.assertRaises(AttributeError):
         _ = st1.col_rename('c', 'C')
     with self.assertRaises(AttributeError):
         _ = st1.col_map({})
     with self.assertRaises(AttributeError):
         _ = st1.col_move_to_back('c')
     with self.assertRaises(AttributeError):
         _ = st1.col_move_to_front('c')
     st1._unlock()
     st1.a = 1
     st1.d = 1
     del st1.b
     self.assertEqual(list(st1.keys()), ['a', 'c', 'd'])
     self.assertEqual(st1.a, 1)
     self.assertEqual(st1.d, 1)
Example #2
0
 def test_col_pop(self):
     cols = ['aa', 'b', 'c', 'μεαν']
     dict1 = {_k: [_i] for _i, _k in enumerate(cols)}
     st1 = Struct(dict1)
     val = st1.col_pop('aa')
     self.assertEqual(val, [0])
     self.assertEqual(list(st1.keys()), cols[1:])
     with self.assertRaises(IndexError):
         st1.col_pop('aa')
     with self.assertRaises(IndexError):
         st1.col_pop(['aa'])
     st2 = st1.col_pop(['b', 'c'])
     self.assertEqual(list(st1.keys()), cols[3:])
     self.assertEqual(list(st2.keys()), ['b', 'c'])
     self.assertEqual(st2.b, [1])
     self.assertEqual(st2.c, [2])
     st3 = st1.col_pop(slice(None))
     self.assertEqual(st1.get_ncols(), 0)
     self.assertEqual(st3.get_ncols(), 1)
     self.assertEqual(st3.μεαν, [3])