code: def test_umda_calculate_probability():
      new_umda = umda.UMDA()
  
      boolean_space = boolean.BooleanSpace(n_agents=5, n_variables=2)
  
      probs = new_umda._calculate_probability(boolean_space.agents)
  
      assert probs.shape == (2, 1)
  
file_path: opytimizer/tests/opytimizer/optimizers/boolean/test_umda.py
imports: import numpy
  from opytimizer.optimizers.boolean import umda
  from opytimizer.spaces import boolean
label_desc: Use image node_id 3 for calling a global function with example usage: test_umda_calculate_probability() without return types