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)
