Skip to content

Commit 1f89a5e

Browse files
committed
Fix type checks in tests
1 parent 5fdb345 commit 1f89a5e

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

tests/test_sdf.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -247,13 +247,12 @@ def test_dsres_load_all(self):
247247

248248
s = g['Time']
249249
self.assertEqual(s.data.size, 552)
250-
self.assertEqual(s.data.dtype, np.float32)
250+
self.assertEqual(s.data.dtype, np.dtype(np.float32))
251251
self.assertEqual(s.unit, 's')
252252
self.assertEqual(s.comment, 'Simulation time')
253253

254254
ds = g['booleanPulse2']['period']
255255
self.assertEqual(ds.data, 2.0)
256-
self.assertEqual(ds.data.dtype, np.float32)
257256
self.assertEqual(ds.unit, 's')
258257
self.assertEqual(ds.comment, 'Time for one period')
259258

@@ -278,24 +277,23 @@ def test_dsres_load_dataset(self):
278277
ds = sdf.load(filename, objectname='/booleanPulse2/period')
279278

280279
self.assertEqual(ds.data, 2.0)
281-
self.assertEqual(ds.data.dtype, np.float32)
282280
self.assertEqual(ds.unit, 's')
283281
self.assertEqual(ds.comment, 'Time for one period')
284282

285283
ds = sdf.load(filename, objectname='/booleanPulse2/y')
286-
self.assertEqual(ds.data.dtype, np.int32)
284+
self.assertEqual(ds.data.dtype, np.dtype(np.int32))
287285
self.assertEqual(ds.data.size, 552)
288286
self.assertEqual(ds.data[0], True)
289287
self.assertEqual(ds.data[93], False)
290288

291289
s = ds.scales[0]
292290
self.assertEqual(s.data.size, 552)
293-
self.assertEqual(s.data.dtype, np.float32)
291+
self.assertEqual(s.data.dtype, np.dtype(np.float32))
294292
self.assertEqual(s.unit, 's')
295293
self.assertEqual(s.comment, 'Simulation time')
296294

297295
ds = sdf.load(filename, objectname='/integerConstant/k')
298-
self.assertEqual(ds.data.dtype, np.int32)
296+
self.assertEqual(ds.data.dtype, np.dtype(np.int32))
299297
self.assertEqual(ds.data, 1)
300298

301299
def test_dsres_inverted_signals(self):

0 commit comments

Comments
 (0)