generic tasks

Author: l

sprint-5/10-predict.py

@@ -0,0 +1,30 @@
+from dataclasses import dataclass
+
+@dataclass(frozen=True)
+class Animal:
+    name: str
+    species: str
+
+@dataclass(frozen=True)
+class Person:
+    name: str
+    age: int
+
+@dataclass(frozen=True)
+class FamilyTree:
+    parent: Person
+    members: list
+
+pet = Animal(name="Gromit", species="Dog")
+fatma = Person(name="Fatma", age=4)
+aisha = Person(name="Aisha", age=6)
+imran = Person(name="Imran", age=30)
+
+family = FamilyTree(parent=imran, members=[fatma, aisha, pet])
+
+def print_family_tree(family: FamilyTree):
+    print(family.parent.name)
+    for child in family.members:
+        print(f"{child.name} ({child.age} years old)")
+
+print_family_tree(family)

sprint-5/11-fix.py

@@ -0,0 +1,63 @@
+from dataclasses import dataclass
+from typing import List
+
+@dataclass(frozen=True)
+class Animal:
+    name: str
+    size: str
+
+@dataclass(frozen=True)
+class Person:
+    name: str
+    age: int
+
+@dataclass(frozen=True)
+class Tree[T]:
+    parent: T
+    children: List[T]
+
+    def print_tree(self):
+        print(self.parent)
+        for child in self.children:
+            print(child)
+
+fatma = Person(name="Fatma", age=4)
+aisha = Person(name="Aisha", age=6)
+imran = Person(name="Imran", age=30)
+family_tree = Tree[Person](parent=imran, children=[fatma, aisha])
+
+cats = Animal(name="Cat", size="Small")
+dogs = Animal(name="Dog", size="Medium")
+mammals = Animal(name="Mammals", size="Variable")
+species_tree = Tree[Animal](parent=mammals, children=[cats, dogs])
+
+family_tree.print_tree()
+species_tree.print_tree()
+
+# Task 11:
+# Experiment with mypy and make sure that the family tree only takes `Person` types and the species tree only takes `Animal` types.
+#
+# We are going to add printing to the above code.
+#
+# Unlike our earlier example, we want to avoid having to create two
+# separate looping methods to print out an entire tree for each datatype.
+# So we have created a single looping function within Tree that will work for any datatype.
+#
+# Currently the `Tree.print_tree()` function doesn't do anything.
+#
+# Add some appropriate methods to each of the Animal and Person classes to allow it to work.
+#
+# **Stretch task**
+#
+# Think of another type of data that can be organised into a tree.
+#
+# Add a new class for this, instantiate some variables, and have the existing `Tree` class print it out.
+# Here is an example of what should be printed out
+'''
+Imran (30 years old)
+- Fatma (4 years old)
+- Aisha (6 years old))
+Mammals (Variable size)
+- Cat (Small size)
+- Dog (Medium size)
+'''