Jessica Walkenhorst commited on
Commit
4ac520e
1 Parent(s): ac562d5
src/maorganizer/datawrangling.py CHANGED
@@ -17,7 +17,7 @@ class Person:
17
  name: str
18
 
19
  def __post_init__(self):
20
- self.name = self.name.title()
21
 
22
  def __hash__(self):
23
  return hash(self.name)
 
17
  name: str
18
 
19
  def __post_init__(self):
20
+ self.name = ' '.join([namepart for namepart in self.name.strip().title().split(' ') if namepart])
21
 
22
  def __hash__(self):
23
  return hash(self.name)
tests/test_datawrangling.py CHANGED
@@ -33,3 +33,11 @@ def test__Person_firstname_is_extracted_correctly(name, firstname):
33
  )
34
  def test_Person_lastname_is_extracted_correctly(name, lastname):
35
  assert Person(name).lastname == lastname
 
 
 
 
 
 
 
 
 
33
  )
34
  def test_Person_lastname_is_extracted_correctly(name, lastname):
35
  assert Person(name).lastname == lastname
36
+
37
+
38
+ def test_whitespace_gets_deleted_from_edges_of_name():
39
+ assert Person(" zaphod beeblebrox ") == Person("zaphod beeblebrox")
40
+
41
+
42
+ def test_multiple_whitespace_gets_correctly_deleted_from_inside_a_name():
43
+ assert Person("Zaphod Beeblebrox") == Person("Zaphod Beeblebrox")