The assertion line tells you that filter_dim_a.prod() and filter_dim_a[0] differ.
For example, you could add a print and check what these two items are equal to. To pass the assertion, they should be equal to each other. Then, try to figure out why they are not. (Using a debugger helps you a lot with the back-tracking of errors, letting you climb the error stack trace.)