MilesCranmer commited on
Commit
fbd0ad8
1 Parent(s): 0e86456

Test empty units as well

Browse files
Files changed (1) hide show
  1. pysr/test/test.py +1 -1
pysr/test/test.py CHANGED
@@ -957,7 +957,7 @@ class TestDimensionalConstraints(unittest.TestCase):
957
  (np.ones((10, 1)), np.ones(10), None, "m/s"),
958
  (np.ones((10, 1)), np.ones(10), None, ["m/s"]),
959
  (np.ones((10, 1)), np.ones((10, 1)), None, ["m/s"]),
960
- (np.ones((10, 1)), np.ones((10, 2)), None, ["m/s", "km"]),
961
  ]
962
  for X, y, X_units, y_units in valid_units:
963
  _check_assertions(
 
957
  (np.ones((10, 1)), np.ones(10), None, "m/s"),
958
  (np.ones((10, 1)), np.ones(10), None, ["m/s"]),
959
  (np.ones((10, 1)), np.ones((10, 1)), None, ["m/s"]),
960
+ (np.ones((10, 1)), np.ones((10, 2)), None, ["m/s", ""]),
961
  ]
962
  for X, y, X_units, y_units in valid_units:
963
  _check_assertions(