def METHOD_NAME(self):
    layer_weights_map = plotting_utils.map_all_module_weights_to_data_frame(model)
    num_conv_and_linear_layers = 0
    for name, module in model.named_modules():
        if isinstance(module, (torch.nn.modules.conv.Conv2d, torch.nn.modules.linear.Linear)):
            num_conv_and_linear_layers += 1
    # verify that there are the same number of data frames as there are conv and linear layers
    self.assertEqual(num_conv_and_linear_layers, len(layer_weights_map))