CSES - Same level

Your task is to count the number of nodes on a given level of a binary tree. The root is at level 1, its children are at level 2, etc.. You may assume that the tree has at most 100 nodes.

The example in the code template represents the following tree:

Here level 1 has one node, level 2 has one node, level 3 has two nodes, and other levels have no nodes.

In a file samelevel.py, implement a function count that returns the number of nodes.

from collections import namedtuple

def count(node, level):
    # TODO

if __name__ == "__main__":
    Node = namedtuple("Node", ["left", "right"])
    tree = Node(None,Node(Node(None,None),Node(None,None)))
    print(count(tree,1)) # 1
    print(count(tree,2)) # 1
    print(count(tree,3)) # 2
    print(count(tree,4)) # 0