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)
def test_col_remove(self):
cols = ['aa', 'b', 'c', 'μεαν']
dict1 = {_k: [_i] for _i, _k in enumerate(cols)}
st1 = Struct(dict1)
self.assertEqual(list(st1.keys()), cols)
st1.col_remove('aa')
self.assertEqual(list(st1.keys()), cols[1:])
with self.assertRaises(IndexError):
st1.col_remove('aa')
st1.col_remove(['b', 'c'])
self.assertEqual(list(st1.keys()), cols[3:])
with self.assertRaises(IndexError):
st1.col_remove(['μεαν', 'b', 'c'])
with self.assertRaises(TypeError):
st1.col_remove(2)
self.assertEqual(list(st1.keys()), cols[3:])
st1.col_remove(['μεαν'])
self.assertEqual(list(st1.keys()), [])
st1.col_remove([])
self.assertEqual(list(st1.keys()), [])