lint

Author: petrelharp

python/tests/test_tree_stats.py

@@ -48,22 +48,25 @@
 # Notes for refactoring:
 #
 # Things we need to test here are:
-# 1. general_stat: correctly uses summary functions
-# 2. general_stat: branch mode, correctness
-# 3. general_stat: site mode, correctness
-# 4. general_stat: node mode, correctness
-# 5. sample sets: correctness
-# 6. indexes: correctness
-# 7. genome windowing: correctness
-# 8. time windowing: correctness
-# 9. dropping dimensions, output
-# 10. span normalise
+# First, for general_stat, AFS, relatedness_matrix, and a few others:
+# 1. branch mode, correctness
+# 2. site mode, correctness
+# 3. node mode, correctness
+# 4. sample sets: correctness
+# 5. indexes: correctness
+# 6. genome windowing: correctness
+# 7. time windowing: correctness
+# 8. dropping dimensions, output
+# 9. span normalise, correctness
+# And, more specifically:
+# 10. general_stat: correctly uses summary functions
 # 11. sample_count_stat: correctly uses summary functions
 # 12. each statistic: a single tree sufficies, with edge cases
 #   a. agrees with naive version, polarised and not;
 #   b. agrees with python version, polarised and not;
 #   c. stat-specific options (eg centre)
 
+
 def cached_np(func):
     """
     Decorator to speed up functions that take numpy arrays as positional

python/tskit/trees.py

@@ -8022,7 +8022,7 @@ def __one_way_sample_set_stat(
             raise ValueError("Sample sets must contain at least one element")
 
         flattened = util.safe_np_int_cast(np.hstack(sample_sets), np.int32)
-        use_tw = (ll_method.__name__ == "allele_frequency_spectrum")
+        use_tw = ll_method.__name__ == "allele_frequency_spectrum"
         if use_tw:
             stat = self.__run_windowed_stat_tw(
                 windows,