Spaces:
Runtime error
Runtime error
Jessica Walkenhorst
commited on
Commit
•
4ac520e
1
Parent(s):
ac562d5
Fix bug
Browse files
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")
|