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")
def test_scale_types(self): pmt = PlotMetricsTracker() pmt.addScaleType("index", int) self.assertEqual(pmt.getType("index"), int) pmt.addScaleType("value", float) self.assertEqual(pmt.getType("value"), float) with self.assertRaises(KeyError): pmt.getScalesForType("index") pmt.setDataTypeKey("TestData1") self.assertEqual(pmt.getScalesForType("index"), (None, None)) self.assertEqual(pmt.getScalesForType("value"), (None, None)) pmt.setDataTypeKey("TestData2") self.assertEqual(pmt.getScalesForType("index"), (None, None)) self.assertEqual(pmt.getScalesForType("value"), (None, None)) pmt.setScalesForType("index", 1, 2) pmt.setScalesForType("value", 1.5, 2.9) self.assertEqual(pmt.getScalesForType("index"), (1, 2)) self.assertEqual(pmt.getScalesForType("value"), (1.5, 2.9)) pmt.setDataTypeKey("TestData1") self.assertEqual(pmt.getScalesForType("index"), (None, None)) self.assertEqual(pmt.getScalesForType("value"), (None, None)) pmt.setScalesForType("index", 10, 20) pmt.setScalesForType("value", 2.5, 3.9) self.assertEqual(pmt.getScalesForType("index"), (10, 20)) self.assertEqual(pmt.getScalesForType("value"), (2.5, 3.9)) with self.assertRaises(KeyError): pmt.setScalesForType("time", 99, 999)