qdscreen/tests/test_core.py

• 1 E501: line too long (124 > 120 characters)
  • 133 dtype=np.int64) # indeed computing parents from adjmat with np.where returns this dtype
• 24 S101: Use of assert detected. The enclosed code will be removed when compiling to optimised byte code.
  • 166 assert qd1.get_roots(names=False) == list(roots_ar)
  • 168 assert qd1.get_roots_with_children(names=False) == list(roots_wc_ar)
  • 171 assert qd1.get_roots(names=True) == list(roots)
  • 172 assert qd1.get_roots_with_children(names=True) == list(roots_wc)
  • 179 assert qd1.get_roots() == qd1.get_roots(names=not is_np)
  • 180 assert qd1.get_roots_with_children() == qd1.get_roots_with_children(names=not is_np)
  • 185 assert qd1.get_arcs_str_list(names=False) == ['1 -> 8', '3 -> 5', '9 -> 1', '9 -> 7']
  • 188 assert qd1._adjmat is None
  • 190 assert qd1.get_arcs_str_list(names=False) == ['9 -> 1', '3 -> 5', '9 -> 7', '1 -> 8']
  • 194 assert qd1.get_arcs_str_list(names=True) == ['b -> i', 'd -> f', 'j -> b', 'j -> h']
  • 197 assert qd1._adjmat is None
  • 199 assert qd1.get_arcs_str_list(names=True) == ['j -> b', 'd -> f', 'j -> h', 'b -> i']
  • 204 assert qd1.get_arcs_str_list() == qd1.get_arcs_str_list(names=not is_np)
  • 231 assert compact_str == "QDForest (10 vars = 6 roots + 4 determined by 2 of the roots)"
  • 236 assert headers_str == """QDForest (10 vars):
  • 240 assert qd1.to_str() == headers_str
  • 245 assert trees_str == u"""
  • 256 assert trees_str == u"""
  • 266 assert full_str == headers_str + u"\n" + trees_str
  • 269 assert full_str.encode('utf-8') == str(qd1)
  • 271 assert full_str == str(qd1)
  • 312 assert Xsel.tolist() == [['A'], ['A'], ['N']]
  • 332 assert list(qd_forest.roots) == ["foo"]
  • 336 assert list(only_important_features_df.columns) == ["foo"]