def test_non_attribute_extraction(self): """Handles data in __init__ that have are not attributes""" e = Extractor().datatype_ext # Arrange expected = [] line = ' self.setup()' # Act actual = e._extract_attribute_defaults(line) # Assert self.assertEqual(expected, actual)
def test_default_attribute_zero_length(self): # Arrange e = Extractor().datatype_ext expected = [] line = " self.teeth" # Act actual = e._extract_attribute_defaults(line) # Assert self.assertEqual(expected, actual)
def test_default_value_extraction(self): """An attribute default value can be extracted from line of code""" # Arrange e = Extractor().datatype_ext expected = 'Sharp' line = " self.teeth = Sharp" # Act actual = e._extract_attribute_defaults(line) # Assert self.assertEqual(expected, actual)