Skip to content

Commit

Permalink
Fixed the build
Browse files Browse the repository at this point in the history
The TestCase.assertEquals() method appears to be available
in Python 3.9 thorugh 3.11, but not in Python 3.12.
  • Loading branch information
vruusmann committed Dec 6, 2024
1 parent e0ee25a commit 7ab3aef
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions sklearn2pmml/cross_reference/tests/test_cross_reference.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,17 +59,17 @@ def test_transform(self):
memorizer = Memorizer(memory, ["int"])
X = numpy.asarray([[-1], [1]])
memorizer.fit(X)
self.assertEquals(0, len(memory))
self.assertEqual(0, len(memory))
Xt = memorizer.transform(X)
self.assertEqual((2, 0), Xt.shape)
self.assertEqual(1, len(memory))
self.assertEqual([-1, 1], memory["int"].tolist())

memory = dict()
self.assertEquals(0, len(memory))
self.assertEqual(0, len(memory))
memorizer = Memorizer(memory, ["int"], transform_only = False)
memorizer.fit(X)
self.assertEquals(1, len(memory))
self.assertEqual(1, len(memory))

memory = DataFrame()
self.assertEqual((0, 0), memory.shape)
Expand Down
2 changes: 1 addition & 1 deletion sklearn2pmml/tests/test_statsmodels.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def test_ordinal_classification(self):
self.assertTrue(hasattr(classifier, "results_"))
self.assertEqual(["c1", "c2", "c3", "c4", "c5"], classifier.classes_.tolist())
self.assertEqual((10, ), classifier.coef_.shape)
self.assertEquals(0, classifier.intercept_)
self.assertEqual(0, classifier.intercept_)
self.assertEqual((4, ), classifier.threshold_.shape)
classifier.remove_data()
pred = classifier.predict(diabetes_X)
Expand Down

0 comments on commit 7ab3aef

Please sign in to comment.