Merge pull request #386 from monicayao/pr376/statsv2

[TST] Additional smoke tests for stats.py

tedana/tests/test_stats_computefeats2.py

@@ -49,3 +49,21 @@ def test_break_computefeats2():
     assert str(e_info.value) == ('Second dimensions (number of volumes) of data ({0}) '
                                  'and mmix ({1}) do not match.'.format(data.shape[0],
                                                                        mmix.shape[0]))
+
+
+# SMOKE TEST
+
+def test_smoke_computefeats2():
+    """
+    Ensures that computefeats2 works with random inputs and different optional parameters
+    """
+    n_samples, n_times, n_components = 100, 20, 6
+    data = np.random.random((n_samples, n_times))
+    mmix = np.random.random((n_times, n_components))  
+    mask = np.random.randint(2, size=n_samples)     
+
+    assert computefeats2(data, mmix) is not None
+    assert computefeats2(data, mmix, mask=mask) is not None
+    assert computefeats2(data, mmix, normalize=False) is not None     
+
+

tedana/tests/test_stats_get_coeffs.py

@@ -76,3 +76,21 @@ def test_break_get_coeffs():
         get_coeffs(data, X, mask, add_const=False)
     assert str(e_info.value) == ('First dimensions of data ({0}) and mask ({1}) do not '
                                  'match'.format(data.shape[0], mask.shape[0]))
+
+
+
+# SMOKE TEST 
+def test_smoke_get_coeffs():
+    """
+    Ensure that get_coeffs returns outputs with different inputs and optional paramters
+    """
+    n_samples, n_echos, n_times, n_components = 100, 5, 20, 6
+    data_2d = np.random.random((n_samples, n_times))
+    #data_3d = np.random.random((n_samples, n_echos, n_times)) 
+    x = np.random.random((n_times, n_components))
+    mask = np.random.randint(2, size=n_samples)     
+
+    assert get_coeffs(data_2d, x) is not None
+    # assert get_coeffs(data_3d, x) is not None TODO: submit an issue for the bug 
+    assert get_coeffs(data_2d, x, mask=mask) is not None
+    assert get_coeffs(data_2d, x, add_const=True) is not None

tedana/tests/test_stats_getfbounds.py

@@ -4,6 +4,7 @@
 
 import numpy as np
 import pytest
+import random
 
 from tedana.stats import getfbounds
 
@@ -13,3 +14,17 @@ def test_getfbounds():
 
     for n_echos in good_inputs:
         getfbounds(n_echos)
+
+
+# SMOKE TEST 
+
+def test_smoke_getfbounds():
+    """ 
+    Ensures that getfbounds returns outputs when fed in a random number of echos
+    """
+    n_echos = random.randint(3, 10) # At least two echos!
+    f05, f025, f01 = getfbounds(n_echos)
+
+    assert f05 is not None
+    assert f025 is not None
+    assert f01 is not None