MilesCranmer commited on
Commit
7a8dabd
1 Parent(s): b3a5026

Trigger #544 in tests

Browse files
Files changed (1) hide show
  1. pysr/test/test.py +2 -2
pysr/test/test.py CHANGED
@@ -1038,7 +1038,7 @@ class TestDimensionalConstraints(unittest.TestCase):
1038
  valid_units = [
1039
  (np.ones((10, 2)), np.ones(10), ["m/s", "s"], "m"),
1040
  (np.ones((10, 1)), np.ones(10), ["m/s"], None),
1041
- (np.ones((10, 1)), np.ones(10), None, "m/s"),
1042
  (np.ones((10, 1)), np.ones(10), None, ["m/s"]),
1043
  (np.ones((10, 1)), np.ones((10, 1)), None, ["m/s"]),
1044
  (np.ones((10, 1)), np.ones((10, 2)), None, ["m/s", ""]),
@@ -1053,7 +1053,7 @@ class TestDimensionalConstraints(unittest.TestCase):
1053
  )
1054
  invalid_units = [
1055
  (np.ones((10, 2)), np.ones(10), ["m/s", "s", "s^2"], None),
1056
- (np.ones((10, 2)), np.ones(10), ["m/s", "s", "s^2"], "m"),
1057
  (np.ones((10, 2)), np.ones((10, 2)), ["m/s", "s"], ["m"]),
1058
  (np.ones((10, 1)), np.ones((10, 1)), "m/s", ["m"]),
1059
  ]
 
1038
  valid_units = [
1039
  (np.ones((10, 2)), np.ones(10), ["m/s", "s"], "m"),
1040
  (np.ones((10, 1)), np.ones(10), ["m/s"], None),
1041
+ (np.ones((10, 1)), np.ones(10), None, "km/s"),
1042
  (np.ones((10, 1)), np.ones(10), None, ["m/s"]),
1043
  (np.ones((10, 1)), np.ones((10, 1)), None, ["m/s"]),
1044
  (np.ones((10, 1)), np.ones((10, 2)), None, ["m/s", ""]),
 
1053
  )
1054
  invalid_units = [
1055
  (np.ones((10, 2)), np.ones(10), ["m/s", "s", "s^2"], None),
1056
+ (np.ones((10, 2)), np.ones(10), ["m/s", "s", "s^2"], "km"),
1057
  (np.ones((10, 2)), np.ones((10, 2)), ["m/s", "s"], ["m"]),
1058
  (np.ones((10, 1)), np.ones((10, 1)), "m/s", ["m"]),
1059
  ]