def test_set_attribute_type_with_sequence_of_attributes(self): dataset = Dataset(self.test_data) dataset.set_attribute_type(AttributeType.IDENTIFYING, "id", "name") self.assertEqual(AttributeType.IDENTIFYING.value, dataset._attributes[0].type.value) self.assertEqual(AttributeType.IDENTIFYING.value, dataset._attributes[1].type.value)
def test_set_attribute_type_with_single_attribute(self): dataset = Dataset(self.test_data) dataset.set_attribute_type(AttributeType.IDENTIFYING, "id") self.assertEqual(AttributeType.IDENTIFYING.value, dataset._attributes[0].type.value)