def test_creation(self): pmt = PlotMetricsTracker() self.assertIsNone(pmt.getType(None)) with self.assertRaises(KeyError): pmt.getType("type") self.assertIsNone(pmt.getDataTypeKey()) pmt.setDataTypeKey("test data type") self.assertEqual(pmt.getDataTypeKey(), "test data type")