Example #1
0
 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)
Example #2
0
 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)