def METHOD_NAME(self):
    head = FeatureHead(backbone_name='slowfast', average_clips='score')
    x_slow = torch.rand(1, 64, 2, 7, 7)
    x_fast = torch.rand(1, 32, 6, 7, 7)
    x = (x_slow, x_fast)
    feat = head(x)
    assert feat.shape == torch.Size([1, 96])
    head = FeatureHead(
        backbone_name='slowfast', spatial_type=None, average_clips='score')
    feat = head(x)
    assert feat.shape == torch.Size([1, 96, 7, 7])
    with pytest.raises(AssertionError):
        head = FeatureHead(
            backbone_name='slowfast',
            temporal_type=None,
            average_clips='score')
        feat = head(x)