def test_overwrite_data(self): db1 = DatasetBuilder(name='db1', data=[1, 2, 3]) msg = "Cannot overwrite data." with self.assertRaisesWith(AttributeError, msg): db1.data = [4, 5, 6]
def test_set_data(self): db1 = DatasetBuilder(name='db1') db1.data = [4, 5, 6] self.assertEqual(db1.data, [4, 5, 6])